update

nguyen-brat · Oct 30, 2023 · e83c20a · e83c20a
1 parent 9fc2913
commit e83c20a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/model/claim_verification/joint_cross_encoder/trainer.py b/model/claim_verification/joint_cross_encoder/trainer.py
@@ -186,7 +186,7 @@ def __call__(
         metrics = [MulticlassF1Score(num_classes=3), MulticlassConfusionMatrix(num_classes=3)]
         for epoch in range(epochs):
             training_steps = 0
-            print(f'epoch: {epoch+1}/{epochs} ')
+            self.accelerator.print(f'epoch: {epoch+1}/{epochs} ')
             self.model.zero_grad()
             self.model.train()
 
@@ -275,7 +275,7 @@ def val_evaluation(self,
                        ):
         with torch.no_grad():
             with torch.cuda.amp.autocast(): ###################
-                print('Val evaluation processing !')
+                self.accelerator.print('Val evaluation processing !')
                 output = []
                 for fact_claims_ids, labels, is_positive, _ in val_dataloader:
                     multi_evident_logits, _ = self.model(fact_claims_ids, is_positive)