Ziyú Ye

Results 1 comments of Ziyú Ye

Ah, I just realized the cause for the KeyError is that `torch.nn.DataParallel(model)` will wrap a module outside the model. A simple fix to avoid this error would be to add...