The new pytorch(0.4) changes its data type, so some fixes need to be done.
The new pytorch(0.4) changes its data type, so some fixes need to be done.