使用nn.init自定義權重，總是報錯RuntimeError: Expected object of scalar type Double but got scalar type Float for

編程小白求大佬拯救！ 最近爲了畢設在做MobileNet的實驗，在訓練網絡的時候通過nn.init自定義了權重，代碼如下： 運行之後系統報錯： 經過調試之後發現原代碼裏面的m.weight的dtype是tensor.float32，於是百度了很多改格式的方法，tensor.type()，tensor.double()啥的都用過了，還是沒有用。有沒有大神可以幫忙指條明路（被自己菜哭了）

>>阅读原文<<