PyTorch神经网络项目文件初始化脚本

Python 脚本

import codecs
import os
from codecs import StreamReaderWriter


init_filename: str = '__init__.py'


def write_to_explanation(explanation_filename: StreamReaderWriter,
                         filename: str, explain: str = "") -> None:
    explanation_filename.write("- {} - {}\n".format(filename, explain))
    print("Successfully create {}.\n".format(filename))


def create_file(filename: str, explanation_filename: StreamReaderWriter, explain: str = "", encoding: str = "utf-8"):
    with open(filename, "w") as _:
        pass
    write_to_explanation(explanation_filename, filename, explain)


def create_folder(filename: str, explanation_filename: StreamReaderWriter, explain: str = "", encoding: str = "utf-8"):
    os.mkdir(filename)
    write_to_explanation(explanation_filename, filename, explain)
    

def create_module(filename: str, explanation_filename: StreamReaderWriter, explain: str = "", encoding: str = "utf-8"):
    create_folder(filename, explanation_filename, explain)
    create_file(f'{ filename }/{ init_filename }', explanation_filename, explain)
    write_to_explanation(explanation_filename, filename, explain)


with codecs.open("EXPLANATION.md", "w", "utf-8") as explanation:
    explanation.write("# EXPLANATION\n")

    create_module("checkpoints", explanation, "存放模型的地方")
    create_module("data", explanation, "定义各种用于训练测试的dataset")
    create_file("eval.py", explanation, "测试代码")
    create_file("loss.py", explanation, "定义各种loss")
    create_file("metrics.py", explanation, "定义各种约定俗成的评估标准")
    create_module("model", explanation, "定义各种实验中的模型,建议每个模型创建一个package")
    create_file("options.py", explanation, "定义各种实验参数,以命令行形式传入")
    create_file("README.md", explanation, "介绍一下自己的repo")
    create_module("scripts", explanation, "各种训练、测试脚本")
    create_file("train.py", explanation, "训练代码")
    create_module("utils", explanation, "各种工具代码")

    explanation.write("参考资料:[Pytorch实验代码的亿些小细节](https://zhuanlan.zhihu.com/p/409662511)\n")


参考资料

Pytorch实验代码的亿些小细节

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值