From e83c20af0794887da3e0dd1eec29979ad21956ff Mon Sep 17 00:00:00 2001 From: nguyen-brat Date: Tue, 31 Oct 2023 02:24:06 +0700 Subject: [PATCH] update --- model/claim_verification/joint_cross_encoder/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model/claim_verification/joint_cross_encoder/trainer.py b/model/claim_verification/joint_cross_encoder/trainer.py index 0c2841f..8492c86 100644 --- 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)