From 4f479d2fe7888a049bddd49e0b4079786d690519 Mon Sep 17 00:00:00 2001 From: Mahmoud Salhab Date: Mon, 5 Sep 2022 09:01:35 +0300 Subject: [PATCH 1/2] Update train.py --- train.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index e8035c1..d83095d 100644 --- a/train.py +++ b/train.py @@ -41,8 +41,9 @@ def load_checkpoint(checkpoint_path, model, optimizer): assert os.path.isfile(checkpoint_path) checkpoint_dict = torch.load(checkpoint_path, map_location='cpu') - iteration = checkpoint_dict['iteration'] - optimizer.load_state_dict(checkpoint_dict['optimizer']) + iteration = checkpoint_dict.get('iteration', 0) + if 'optimizer' in checkpoint_dict: + optimizer.load_state_dict(checkpoint_dict['optimizer']) model_for_loading = checkpoint_dict['model'] model.load_state_dict(model_for_loading.state_dict()) print("Loaded checkpoint '{}' (iteration {})" .format( From 165be756566b11ae5fe570a3a5c73908cb5c9e09 Mon Sep 17 00:00:00 2001 From: Mahmoud Salhab Date: Sun, 13 Nov 2022 10:39:12 +0300 Subject: [PATCH 2/2] Update requirements.txt --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 20c8b3e..49c737c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ matplotlib==2.1.0 tensorflow numpy==1.13.3 inflect==0.2.5 -librosa==0.6.0 +librosa==0.6.2 scipy==1.0.0 tensorboardX==1.1 Unidecode==1.0.22