sierkinhane 4 年之前
父节点
当前提交
04e34ec3fb
共有 1 个文件被更改,包括 4 次插入0 次删除
  1. 4 0
      lib/config/OWN_config.yaml

+ 4 - 0
lib/config/OWN_config.yaml

@@ -37,6 +37,10 @@ TRAIN:
   NESTEROV: False
   RMSPROP_ALPHA:
   RMSPROP_CENTERED:
+  FINETUNE:
+    IS_FINETUNE: true
+    FINETUNE_CHECKPOINIT: 'output/checkpoints/mixed_second_finetune_acc_97P7.pth'
+    FREEZE: true
 
 TEST:
   BATCH_SIZE_PER_GPU: 16