From 832b1cc262c19cc084287b4be3bba1e18b4f1be0 Mon Sep 17 00:00:00 2001 From: Derrick Schultz Date: Mon, 23 Nov 2020 16:12:07 -0500 Subject: [PATCH 1/2] bump albumentations for colab --- scripts/install.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scripts/install.sh b/scripts/install.sh index ea1569d2..cfb29549 100644 --- a/scripts/install.sh +++ b/scripts/install.sh @@ -25,7 +25,8 @@ pip install jupyterlab --upgrade pip install scikit-image tqdm wget pip install cython pyyaml lmdb pip install opencv-python opencv-contrib-python -pip install open3d albumentations requests +pip install open3d requests +pip install albumentations>=5.1.0 pip install qimage2ndarray pip install imageio-ffmpeg pip install face-alignment dlib From 1ac4784abb789ed23efcfbf40340e64f9b35c35a Mon Sep 17 00:00:00 2001 From: Derrick Date: Mon, 7 Dec 2020 16:44:35 -0500 Subject: [PATCH 2/2] support resuming training --- train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/train.py b/train.py index 6ef618f1..cd6b1da2 100644 --- a/train.py +++ b/train.py @@ -24,6 +24,7 @@ def parse_args(): parser.add_argument('--seed', type=int, default=0, help='Random seed.') parser.add_argument('--local_rank', type=int, default=0) parser.add_argument('--single_gpu', action='store_true') + parser.add_argument('--resume', action='store_true') parser.add_argument('--num_workers', type=int) args = parser.parse_args() return args @@ -61,7 +62,7 @@ def main(): sch_G, sch_D, train_data_loader, val_data_loader) current_epoch, current_iteration = trainer.load_checkpoint( - cfg, args.checkpoint) + cfg, args.checkpoint,resume=args.resume) # Start training. for epoch in range(current_epoch, cfg.max_epoch):