Changing minor stuff for pytorch1.0 compatibility

hassony2 / kinetics_i3d_pytorch

Inflated i3d network with inception backbone, weights transfered from tensorflow

MIT License

523 stars 114 forks source link

Closed TheShadow29 closed 5 years ago

TheShadow29 commented 5 years ago

variable.data[0] is deprecated for scalar, changed it to variable.data.item().

hassony2 commented 5 years ago

Thanks ! :)