Closed fclearner closed 5 months ago
看报错好像是writer = SummaryWriter(tensorboard_dir)代码出错,你修改train.py代码了吗?训练语种识别的时候一般不会挂掉,要么你重新git clone尝试一下?
看报错好像是writer = SummaryWriter(tensorboard_dir)代码出错,你修改train.py代码了吗?训练语种识别的时候一般不会挂掉,要么你重新git clone尝试一下?
嗯嗯,我确实加了一个日志打印的代码,我没怀疑到这上面,感谢大佬,我先去调试看看
大佬们,你们在训练语种识别的时候会不会老是挂掉,我这边每训练一轮就挂一次,我尝试改了:
(1)timeout增大:dist.init_process_group(backend='nccl', timeout=datetime.timedelta(seconds=7200000)); (2)batch_size调小; (2)num_workers调成0; 还是报错了:
python3.8 torch版本是2.2.1+cu118 cuda11.6