僅作為記錄,大佬請跳過。
用
loss.requires_grad_(True)
是因為loss沒有設定梯度(是以不能反向傳播)(loss的資料類型是tensor)
設定梯度後即可,展示:
參考
pytorch官方教程 傳送門
僅作為記錄,大佬請跳過。
用
loss.requires_grad_(True)
是因為loss沒有設定梯度(是以不能反向傳播)(loss的資料類型是tensor)
設定梯度後即可,展示:
參考
pytorch官方教程 傳送門