No, there is no parameter equals to 11 . The issue is not the metrics because I’ve observed that it freezes on self.optimizer.step(), I’ve also tried to avoid computing the metrics completely.
About batch_size I’ve tried 32 and 50 . It seems that with 32 the freeze is a little shorter. I’m away from computer now but if I’m not mistaken I’m using Pytorch 0.4.x. Could it be a bug and should I try to update Pytorch?
Thanks!