update script

ai4ce · Nov 7, 2024 · 8115c8d · 8115c8d
1 parent b151281
commit 8115c8d
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/pl_modules/urbannav_jepa_module.py b/pl_modules/urbannav_jepa_module.py
@@ -300,8 +300,6 @@ def compute_loss(self, wp_pred, arrive_pred, feature_pred, feature_gt, batch):
         # Loss is 1 - cos_sim
         direction_loss = 1 - cos_sim.mean()
 
-        print(f"Feature loss: {feature_loss:.4f}, Waypoints loss: {wp_loss:.4f}, Arrived loss: {arrived_loss:.4f}, Direction loss: {direction_loss:.4f}")
-
         return {'waypoints_loss': wp_loss, 'arrived_loss': arrived_loss, 'direction_loss': direction_loss, 'feature_loss': feature_loss}
 
     def compute_loss_polar(self, wp_pred_euclidean, distance_pred, angle_pred, arrive_pred, batch):