remove params overwrite
This commit is contained in:
@@ -19,6 +19,18 @@ data:
|
||||
split: test
|
||||
download: true
|
||||
|
||||
optimization:
|
||||
lr: 0.01
|
||||
weight_decay: 5.0e-4
|
||||
optimizer: adamw
|
||||
epochs: 300
|
||||
freeze_weights: true
|
||||
lr_schedule: cosine
|
||||
warmup: 0
|
||||
momentum: 0.99
|
||||
start_lr: 0.0
|
||||
final_lr: 0.0
|
||||
|
||||
logging:
|
||||
write_tag: iwildcam_test
|
||||
auto_folder: true
|
||||
|
||||
@@ -110,8 +110,6 @@ def main(args):
|
||||
params_path = os.path.join(folder, "params-eval.yaml")
|
||||
with open(params_path, "w") as f:
|
||||
yaml.dump(args, f)
|
||||
with open(os.path.join(folder, "params.yaml"), "w") as f:
|
||||
yaml.dump(args, f)
|
||||
|
||||
model_name = meta_args["model_name"]
|
||||
embed_dim = meta_args["embed_dim"]
|
||||
|
||||
Reference in New Issue
Block a user