整理Pytorch代碼自己的寫作習慣

自己不是純代碼爲主,而主要研究算法,因此自己以前全部代碼都是停留在能用的階段,但自從git clone SRPytorch 這一工程後,覺得需要提升自己代碼可讀性,也便於自己算法的對比。現大體歸納爲以下幾點: 1、適當的目錄 之前很不好的習慣是全部代碼放在一處,甚至放在一個文件中。受SRPytorch的啓發,按模型構建、模型保存、附加類庫等分類,構建適當的目錄,如下圖 這樣工程下只放一個main.
相關文章
相關標籤/搜索