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...
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...