Update experiment/transfer_learning_train&test.py
Browse files
experiment/transfer_learning_train&test.py
CHANGED
|
@@ -39,11 +39,13 @@ def parse_args():
|
|
| 39 |
|
| 40 |
args = parse_args()
|
| 41 |
|
| 42 |
-
dict_to_freeze = {"
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
|
|
|
| 46 |
}
|
|
|
|
| 47 |
layers_to_freeze = dict_to_freeze[args.run]
|
| 48 |
##--------------------------------------------/USER DEFINE
|
| 49 |
|
|
|
|
| 39 |
|
| 40 |
args = parse_args()
|
| 41 |
|
| 42 |
+
dict_to_freeze = {"From_Scratch": 0,
|
| 43 |
+
"Finetuning": 0,
|
| 44 |
+
"freeze_[P1-P3]": 4,
|
| 45 |
+
"freeze_Backbone": 9,
|
| 46 |
+
"freeze_[P1-23]": 23,
|
| 47 |
}
|
| 48 |
+
|
| 49 |
layers_to_freeze = dict_to_freeze[args.run]
|
| 50 |
##--------------------------------------------/USER DEFINE
|
| 51 |
|