pytorch文件名詳解

模型定義 數據處理和加載 訓練模型(Train&Validate) 訓練過程的可視化 測試(Test/Inference)python checkpoints/: 用於保存訓練好的模型,可以使程序在異常退出後仍能從新載入模型,恢復訓練 data/:數據相關操做,包括數據預處理、dataset實現等 models/:模型定義,能夠有多個模型,例如上面的AlexNet和ResNet34,一個模型對應一
相關文章
相關標籤/搜索