Hello!
I’m using nn.Dataparallel and tried to get its name by model.class.base
but it does not work with nn.Dataparallel which only gives string (DataParallel)
is there any solution without changing the model’s status?