Merge pull request #36 from kuzand/patch-1

Update trainer.py
ThibaultGROUEIX · Aug 4, 2021 · e138f49 · e138f49
2 parents e840520 + 0dc0ca8
commit e138f49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/training/trainer.py b/training/trainer.py
@@ -143,7 +143,7 @@ def test_iteration(self):
 
         self.log.update("loss_val_Deformation_ChamferL2", loss_val_Deformation_ChamferL2)
         print(
-            '\r' + colored('[%d: %d/%d]' % (self.epoch, self.iteration, self.len_dataset_test / (self.opt.batch_size)),
+            '\r' + colored('[%d: %d/%d]' % (self.epoch, self.iteration, len(self.dataloader_test)),
                            'red') +
             colored('loss_val_Deformation_ChamferL2:  %f' % loss_val_Deformation_ChamferL2.item(), 'yellow'),
             end='')