From 006b122be2eae6cb411ab4faa07a223ade452c7b Mon Sep 17 00:00:00 2001 From: Fanis Khafizov Date: Thu, 30 Jan 2025 23:01:46 +0300 Subject: [PATCH] update --- code/ResNet/train.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/code/ResNet/train.py b/code/ResNet/train.py index 35afd0d..c694a49 100644 --- a/code/ResNet/train.py +++ b/code/ResNet/train.py @@ -31,8 +31,6 @@ def train(model, optimizer, compressor, criterion, train_dataset, val_dataset, n _, predicted = outputs.max(1) total += targets.size(0) correct += predicted.eq(targets).sum().item() - - tqdm.write(f'Batch {batch_idx+1}/{len(train_dataset)} - Loss: {loss.item():.3f}, Acc: {100. * correct / total:.3f}%') train_loss /= len(train_dataset) train_accuracy = 100. * correct / total