記錄學習pytorch的過程，持續更。
pytorch 減小視訊記憶體消耗，優化視訊記憶體使用，避免out of memory
參考這個比較全：
原因1：在呼叫函式的時候，如果函式返回值是5個，而你只寫了3個的話，也會報這個錯誤，所以將所以的返回值都寫出來，不管你用還是不用。
原因2：是否在遍歷字典的時候沒有使用items；dict = {}
for i in dict.items():
pass
pytorch使用記錄
1.resnet需要載入預訓練模型，但是最後一層fc層的輸出數不是1000。解決方案 先載入預訓練模型，然後取除了最後一層的所有層，再加一層自己需要的在後面。2.model.children 和model.modules 的區別。model.modules 和model.children 均為迭代器...
PyTorch 使用中注意事項
1.把label要轉成longtensor格式 self.y torch.longtensor y 完整使用 如下 1 class imgdataset dataset 2def init self,x,y none,transform none 3 self.x x4 label is requi...
pytorch的基礎記錄
detach torch.abs input,out none 計算輸入張量的每個元素的絕對值 torch.acos input,out none 返回乙個新張量，包含輸入張量每個元素的反余弦函式 torch.add input,value,out none 對輸入張量input逐元素加上標量值va...