From 00712408390f347b061d4749d2bc4bfbeb269063 Mon Sep 17 00:00:00 2001 From: Javi Ribera Date: Sun, 4 Feb 2018 15:21:01 -0500 Subject: [PATCH] no need to print architecture for inference Former-commit-id: 8febd95142bb2b74a2c034e54889b752371af127 --- deliverable/locate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deliverable/locate.py b/deliverable/locate.py index 4e13096..b6e94d2 100644 --- 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)