一些小Trick(pip清華鏡像之類)隨時補充

Pip使用清華鏡像

可以在使用pip的時候加參數-i https://pypi.tuna.tsinghua.edu.cn/simple
例如:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy,這樣就會從清華這邊的鏡像去安裝numpy庫。

Pytorch模型存儲、訓練中間數據存儲爲csv形式等

中間數據存儲

cost_list = []
cost_list.append[cost]
import pandas as pd
model_cost = pd.Series(cost_list)
model_cost.to_csv('./cost.csv')

模型存儲與加載

torch.save(model,'./model/AlexNet.pth')
alex = torch.load('./data/AlexNet.pth')
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章