sierkinhane il y a 4 ans
Parent
commit
04e34ec3fb
1 fichiers modifiés avec 4 ajouts et 0 suppressions
  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