#将baseline程序is_train改成True就是启动训练,此版本训练是多gpu的,如果但gpu训练可以稍微改下
python baseline就能启动训练
#模型预测就是is_train改成False然后把gpu指定改成单卡就可以进行推理了。
python baseline能启动推理
自动调用model_predict函数生成result.tsv文件,支持修改推理不同的文件。
-
Notifications
You must be signed in to change notification settings - Fork 3
core-power/2021CCF-IQIYI
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published