为什么在训练的时候一直卡在这里,希望大佬解答一下,小白刚入门
2条回答 默认 最新
- 爱晚乏客游 2021-04-16 00:18关注
你是一直卡在这里吗还是有一行一行的显示这样?看看runs/train/exp_num/下面有没有一些数据生成,如果有,那就说明说在跑了,只不过你这图片太少,batch大点两三下就计算完了显示不过来而已。如果没有的话,目测是配置问题。参数里面有个worker数目,默认是6,你看你cpu多少核的,改的比核心数小2试试看。但是你这图片这么少,不应该配置不够啊。
parser.add_argument('--workers', type=int, default=6, help='maximum number of dataloader workers')
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 1无用