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')