LYHjunior / Fudan-DATA620004-Projects

Fudan DATA620004 Projects

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Fudan-DATA620004-Projects

训练和测试步骤:直接运行main.py文件,即可进行训练。程序运行结束后可输出训练结果,得到训练好的模型,以及准确率、损失曲线、参数可视化图。

若不进行训练,仅加载训练好的模型进行测试,则需把args.resume参数改成'model.pkl',然后运行main.py文件。

About

Fudan DATA620004 Projects