no need to print architecture for inference

Former-commit-id: 8febd95142bb2b74a2c034e54889b752371af127
javiribera · Feb 4, 2018 · 0071240 · 0071240
1 parent 947a8a8
commit 0071240
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deliverable/locate.py b/deliverable/locate.py
@@ -156,7 +156,7 @@ def __getitem__(self, idx):
 
 # Loss function
 l1_loss = nn.L1Loss()
-criterion_training = losses.ModifiedChamferLoss(height=height, width=width,
+criterion_training = losses.WeightedHausdorffDistance(height=height, width=width,
                                                 return_2_terms=True)
 
 # Restore saved checkpoint (model weights)
@@ -192,7 +192,7 @@ def __getitem__(self, idx):
     # Load model in checkpoint
     model.load_state_dict(checkpoint['model'])
     print("╰─ loaded checkpoint '{}'".format(args.model))
-    print(model)
+    # print(model)
 else:
     print("╰─ E: no checkpoint found at '{}'".format(args.model))
     exit(-1)