validation loss log enable for every epoch

src/train_egnn.py

@@ -909,8 +909,9 @@ def train_model(model, train_loader, val_loader, num_epochs, learning_rate, devi
         train_loss = train_one_epoch(model, train_loader, optimizer, device, use_pointnet, log_interval, beta)
 
         # Validate every few epochs (e.g., every 5 epochs)
-        if (epoch + 1) % 5 == 0:
-            val_loss = validate(model, val_loader, device, use_pointnet)
+        if (epoch + 1) % 1 == 0:
+            val_loss, val_pose_loss, val_corr_loss = validate(model, val_loader, device, use_pointnet)
+            print(val_loss)
             print(f'Epoch {epoch + 1}/{num_epochs} - Training Loss: {train_loss:.4f}, Validation Loss: {val_loss:.4f}')
         else:
             print(f'Epoch {epoch + 1}/{num_epochs} - Training Loss: {train_loss:.4f}')