Shen TingJia

icon company University of Science and Technology of China icon location Anhui Hefei, China

Results 2 issues of Shen TingJia

请问如何读取checkpoint继续训练？

2 comment

好像不支持多卡并行？

好像不支持多卡并行？把model.float().cuda()更改为nn.parallel之后还是会在customTrainer那里爆CUDA:0out of memory