Merge pull request #314 from yoshitomo-matsubara/dev

Support src_ckpt and dst_ckpt for initialization and saving checkpoints respectively

configs/official/coco2017/yoshitomo-matsubara/rrpr2020/ghnd-custom_fasterrcnn_resnet50_fpn_from_fasterrcnn_resnet50_fpn.yaml

@@ -24,8 +24,7 @@ models:
       progress: True
       num_classes: 91
       pretrained_backbone: True
-    experiment: &teacher_experiment !join [*dataset_name, '-', *teacher_model_key]
-    ckpt: !join ['./resource/ckpt/coco2017/teacher/', *teacher_experiment, '.pt']
+    src_ckpt:
   student_model:
     key: &student_model_key 'custom_fasterrcnn_resnet_fpn'
     kwargs:
@@ -45,7 +44,8 @@ models:
             compressor:
             decompressor:
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key, '_from_', *teacher_model_key]
-    ckpt: !join ['./coco/ghnd/', *student_experiment, '.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./coco/ghnd/', *student_experiment, '.pt']
 
 train:
   log_freq: 1000

configs/official/coco2017/yoshitomo-matsubara/rrpr2020/ghnd-custom_maskrcnn_resnet50_fpn_from_maskrcnn_resnet50_fpn.yaml

@@ -24,8 +24,7 @@ models:
       progress: True
       num_classes: 91
       pretrained_backbone: True
-    experiment: &teacher_experiment !join [*dataset_name, '-', *teacher_model_key]
-    ckpt: !join ['./resource/ckpt/coco2017/teacher/', *teacher_experiment, '.pt']
+    src_ckpt:
   student_model:
     key: &student_model_key 'custom_maskrcnn_resnet_fpn'
     kwargs:
@@ -45,7 +44,8 @@ models:
             compressor:
             decompressor:
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key, '_from_', *teacher_model_key]
-    ckpt: !join ['./coco/ghnd/', *student_experiment, '.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./coco/ghnd/', *student_experiment, '.pt']
 
 train:
   log_freq: 1000

configs/official/ilsvrc2012/yoshitomo-matsubara/rrpr2020/at-resnet18_from_resnet34.yaml

@@ -43,15 +43,15 @@ models:
     kwargs:
       num_classes: 1000
       pretrained: True
-    experiment: &teacher_experiment !join [*dataset_name, '-', *teacher_model_key]
-    ckpt: !join ['./resource/ckpt/ilsvrc2012/teacher/', *teacher_experiment, '.pt']
+    src_ckpt:
   student_model:
     key: &student_model_key 'resnet18'
     kwargs:
       num_classes: 1000
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key, '_from_', *teacher_model_key]
-    ckpt: !join ['./imagenet/at/', *student_experiment, '.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./imagenet/at/', *student_experiment, '.pt']
 
 train:
   log_freq: 1000

configs/official/ilsvrc2012/yoshitomo-matsubara/rrpr2020/crd-resnet18_from_resnet34.yaml

@@ -43,15 +43,15 @@ models:
     kwargs:
       num_classes: 1000
       pretrained: True
-    experiment: &teacher_experiment !join [*dataset_name, '-', *teacher_model_key]
-    ckpt: !join ['./resource/ckpt/ilsvrc2012/teacher/', *teacher_experiment, '.pt']
+    src_ckpt:
   student_model:
     key: &student_model_key 'resnet18'
     kwargs:
       num_classes: 1000
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key, '_from_', *teacher_model_key]
-    ckpt: !join ['./imagenet/crd/', *student_experiment, '.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./imagenet/crd/', *student_experiment, '.pt']
 
 train:
   log_freq: 1000

configs/official/ilsvrc2012/yoshitomo-matsubara/rrpr2020/cse_l2-resnet18_from_resnet34.yaml

@@ -43,15 +43,15 @@ models:
     kwargs:
       num_classes: 1000
       pretrained: True
-    experiment: &teacher_experiment !join [*dataset_name, '-', *teacher_model_key]
-    ckpt: !join ['./resource/ckpt/ilsvrc2012/teacher/', *teacher_experiment, '.pt']
+    src_ckpt:
   student_model:
     key: &student_model_key 'resnet18'
     kwargs:
       num_classes: 1000
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key, '_from_', *teacher_model_key]
-    ckpt: !join ['./imagenet/cse_l2/', *student_experiment, '.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./imagenet/cse_l2/', *student_experiment, '.pt']
 
 train:
   log_freq: 1000

configs/official/ilsvrc2012/yoshitomo-matsubara/rrpr2020/ft-resnet18_from_resnet34.yaml

@@ -43,15 +43,15 @@ models:
     kwargs:
       num_classes: 1000
       pretrained: True
-    experiment: &teacher_experiment !join [*dataset_name, '-', *teacher_model_key]
-    ckpt: !join ['./resource/ckpt/ilsvrc2012/teacher/', *teacher_experiment, '.pt']
+    src_ckpt:
   student_model:
     key: &student_model_key 'resnet18'
     kwargs:
       num_classes: 1000
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key, '_from_', *teacher_model_key]
-    ckpt: !join ['./imagenet_1gpu/ft_normal/', *student_experiment, '.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./imagenet_1gpu/ft_normal/', *student_experiment, '.pt']
 
 train:
   log_freq: 1000

configs/official/ilsvrc2012/yoshitomo-matsubara/rrpr2020/kd-resnet18_from_resnet34.yaml

@@ -43,15 +43,15 @@ models:
     kwargs:
       num_classes: 1000
       pretrained: True
-    experiment: &teacher_experiment !join [*dataset_name, '-', *teacher_model_key]
-    ckpt: !join ['./resource/ckpt/ilsvrc2012/teacher/', *teacher_experiment, '.pt']
+    src_ckpt:
   student_model:
     key: &student_model_key 'resnet18'
     kwargs:
       num_classes: 1000
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key, '_from_', *teacher_model_key]
-    ckpt: !join ['./imagenet/kd/', *student_experiment, '.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./imagenet/kd/', *student_experiment, '.pt']
 
 train:
   log_freq: 1000

configs/official/ilsvrc2012/yoshitomo-matsubara/rrpr2020/pad_l2-resnet18_from_resnet34.yaml

@@ -43,15 +43,15 @@ models:
     kwargs:
       num_classes: 1000
       pretrained: True
-    experiment: &teacher_experiment !join [*dataset_name, '-', *teacher_model_key]
-    ckpt: !join ['./resource/ckpt/ilsvrc2012/teacher/', *teacher_experiment, '.pt']
+    src_ckpt:
   student_model:
     key: &student_model_key 'resnet18'
     kwargs:
       num_classes: 1000
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key, '_from_', *teacher_model_key]
-    ckpt: !join ['./imagenet/pad/', *student_experiment, '.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./imagenet/pad/', *student_experiment, '.pt']
 
 train:
   log_freq: 1000

configs/official/ilsvrc2012/yoshitomo-matsubara/rrpr2020/sskd-resnet18_from_resnet34.yaml

@@ -43,15 +43,15 @@ models:
     kwargs:
       num_classes: 1000
       pretrained: True
-    experiment: &teacher_experiment !join [*dataset_name, '-', *teacher_model_key]
-    ckpt: !join ['./resource/ckpt/ilsvrc2012/teacher/', *teacher_experiment, '.pt']
+    src_ckpt:
   student_model:
     key: &student_model_key 'resnet18'
     kwargs:
       num_classes: 1000
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key, '_from_', *teacher_model_key]
-    ckpt: !join ['./imagenet/sskd/', *student_experiment, '.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./imagenet/sskd/', *student_experiment, '.pt']
 
 train:
   log_freq: 1000

configs/official/ilsvrc2012/yoshitomo-matsubara/rrpr2020/tfkd-resnet18_from_resnet18.yaml

@@ -43,15 +43,15 @@ models:
     kwargs:
       num_classes: 1000
       pretrained: True
-    experiment: &teacher_experiment !join [*dataset_name, '-', *teacher_model_key]
-    ckpt: !join ['./resource/ckpt/ilsvrc2012/teacher/', *teacher_experiment, '.pt']
+    src_ckpt:
   student_model:
     key: &student_model_key 'resnet18'
     kwargs:
       num_classes: 1000
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key, '_from_', *teacher_model_key]
-    ckpt: !join ['./imagenet/tfkd/', *student_experiment, '.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./imagenet/tfkd/', *student_experiment, '.pt']
 
 train:
   log_freq: 1000

configs/sample/cifar10/ce/densenet_bc_k12_depth100-final_run.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/densenet_bc_k12_depth100-hyperparameter_tuning.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/densenet_bc_k24_depth250-final_run.yaml

@@ -51,7 +51,8 @@ models:
       memory_efficient: False
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/densenet_bc_k24_depth250-hyperparameter_tuning.yaml

@@ -51,7 +51,8 @@ models:
       memory_efficient: False
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/densenet_bc_k40_depth190-final_run.yaml

@@ -51,7 +51,8 @@ models:
       memory_efficient: False
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/densenet_bc_k40_depth190-hyperparameter_tuning.yaml

@@ -51,7 +51,8 @@ models:
       memory_efficient: False
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/resnet110-final_run.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/resnet110-hyperparameter_tuning.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/resnet1202-final_run.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/resnet1202-hyperparameter_tuning.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/resnet20-final_run.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/resnet20-hyperparameter_tuning.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/resnet32-final_run.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/resnet32-hyperparameter_tuning.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/resnet44-final_run.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/resnet44-hyperparameter_tuning.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/resnet56-final_run.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/resnet56-hyperparameter_tuning.yaml

@@ -50,7 +50,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/wide_resnet16_8-final_run.yaml

@@ -51,7 +51,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-final_run.pt']
 
 train:
   log_freq: 100

configs/sample/cifar10/ce/wide_resnet16_8-hyperparameter_tuning.yaml

@@ -51,7 +51,8 @@ models:
       num_classes: 10
       pretrained: False
     experiment: &student_experiment !join [*dataset_name, '-', *student_model_key]
-    ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
+    src_ckpt:
+    dst_ckpt: !join ['./resource/ckpt/', *dataset_name, '/ce/', *student_experiment, '-hyperparameter_tuning.pt']
 
 train:
   log_freq: 100