diff --git a/trainer/Trainer.py b/trainer/Trainer.py index 54d2654..772f08c 100755 --- a/trainer/Trainer.py +++ b/trainer/Trainer.py @@ -3,8 +3,6 @@ import os import time import copy import psutil -from tqdm import tqdm - import torch from lib.logger import get_logger from lib.loss_function import all_metrics