Allow for resume finetuning from a checkpoint

configs/Kinetics/SLOWFAST_4x16_R50.yaml

@@ -5,6 +5,8 @@ TRAIN:
   EVAL_PERIOD: 10
   CHECKPOINT_PERIOD: 5
   AUTO_RESUME: True
+  CHECKPOINT_FILE_PATH: /path/to/checkpoint
+  FINETUNE: False
 DATA:
   PATH_TO_DATA_DIR: /path/to/kinetics/
   NUM_FRAMES: 32

configs/Kinetics/SLOWFAST_8x8_R101.yaml

@@ -5,6 +5,8 @@ TRAIN:
   EVAL_PERIOD: 10
   CHECKPOINT_PERIOD: 5
   AUTO_RESUME: True
+  CHECKPOINT_FILE_PATH: /path/to/checkpoint
+  FINETUNE: False
 DATA:
   PATH_TO_DATA_DIR:  /path/to/kinetics/
   NUM_FRAMES: 32

configs/Kinetics/SLOWFAST_8x8_R50.yaml

@@ -5,6 +5,8 @@ TRAIN:
   EVAL_PERIOD: 10
   CHECKPOINT_PERIOD: 5
   AUTO_RESUME: True
+  CHECKPOINT_FILE_PATH: /path/to/checkpoint
+  FINETUNE: False
 DATA:
   PATH_TO_DATA_DIR: /path/to/kinetics/
   NUM_FRAMES: 32

configs/Kinetics/TimeSformer_divST_16x16_448.yaml

@@ -5,6 +5,8 @@ TRAIN:
   EVAL_PERIOD: 5
   CHECKPOINT_PERIOD: 5
   AUTO_RESUME: True
+  CHECKPOINT_FILE_PATH: /path/to/checkpoint
+  FINETUNE: False
 DATA:
   PATH_TO_DATA_DIR: /path/to/kinetics/
   NUM_FRAMES: 16

configs/Kinetics/TimeSformer_divST_8x32_224.yaml

@@ -5,6 +5,8 @@ TRAIN:
   EVAL_PERIOD: 5
   CHECKPOINT_PERIOD: 5
   AUTO_RESUME: True
+  CHECKPOINT_FILE_PATH: /path/to/checkpoint
+  FINETUNE: False
 DATA:
   PATH_TO_DATA_DIR: /path/to/kinetics/
   NUM_FRAMES: 8

configs/Kinetics/TimeSformer_divST_8x32_224_4gpus.yaml

@@ -5,6 +5,8 @@ TRAIN:
   EVAL_PERIOD: 5
   CHECKPOINT_PERIOD: 5
   AUTO_RESUME: True
+  CHECKPOINT_FILE_PATH: /path/to/checkpoint
+  FINETUNE: False
 DATA:
   PATH_TO_DATA_DIR: /path/to/kinetics/
   NUM_FRAMES: 8

configs/Kinetics/TimeSformer_divST_96x4_224.yaml

@@ -5,6 +5,8 @@ TRAIN:
   EVAL_PERIOD: 5
   CHECKPOINT_PERIOD: 5
   AUTO_RESUME: True
+  CHECKPOINT_FILE_PATH: /path/to/checkpoint
+  FINETUNE: False
 DATA:
   PATH_TO_DATA_DIR: /path/to/kinetics/
   NUM_FRAMES: 96

configs/Kinetics/TimeSformer_jointST_8x32_224.yaml

@@ -5,6 +5,8 @@ TRAIN:
   EVAL_PERIOD: 5
   CHECKPOINT_PERIOD: 5
   AUTO_RESUME: True
+  CHECKPOINT_FILE_PATH: /path/to/checkpoint
+  FINETUNE: False
 DATA:
   PATH_TO_DATA_DIR: /path/to/kinetics/
   NUM_FRAMES: 8

configs/Kinetics/TimeSformer_spaceOnly_8x32_224.yaml

@@ -5,6 +5,8 @@ TRAIN:
   EVAL_PERIOD: 5
   CHECKPOINT_PERIOD: 5
   AUTO_RESUME: True
+  CHECKPOINT_FILE_PATH: /path/to/checkpoint
+  FINETUNE: False
 DATA:
   PATH_TO_DATA_DIR: /path/to/kinetics/
   NUM_FRAMES: 8

tools/train_net.py

@@ -415,7 +415,7 @@ def train(cfg):
     optimizer = optim.construct_optimizer(model, cfg)
 
     # Load a checkpoint to resume training if applicable.
-    if not cfg.TRAIN.FINETUNE:
+    if cfg.TRAIN.FINETUNE:
       start_epoch = cu.load_train_checkpoint(cfg, model, optimizer)
     else:
       start_epoch = 0