该仓库为 Yas 的模型训练程序
假设你会设置基本的pytorch环境。
- 生成数据集
python main.py gen
- 训练
python main.py train
在mona/config.py
处可以修改数据集大小、epoch等参数
如果要加速大数据集的生成,可以使用pargen.py
如果想使用大数据集验证(又不想大量占用硬盘和内存),可以使用onlineval.py
来进行在线验证
- Issue
- QQ群:801106595
该仓库为 Yas 的模型训练程序
假设你会设置基本的pytorch环境。
python main.py gen
python main.py train
在mona/config.py
处可以修改数据集大小、epoch等参数
如果要加速大数据集的生成,可以使用pargen.py
如果想使用大数据集验证(又不想大量占用硬盘和内存),可以使用onlineval.py
来进行在线验证