remove params overwrite

This commit is contained in:
Yann Ahlgrim
2026-06-19 11:24:08 +02:00
parent 3aef22df62
commit 99489c52dd
2 changed files with 12 additions and 2 deletions
+12
View File
@@ -19,6 +19,18 @@ data:
split: test split: test
download: true 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: logging:
write_tag: iwildcam_test write_tag: iwildcam_test
auto_folder: true auto_folder: true
-2
View File
@@ -110,8 +110,6 @@ def main(args):
params_path = os.path.join(folder, "params-eval.yaml") params_path = os.path.join(folder, "params-eval.yaml")
with open(params_path, "w") as f: with open(params_path, "w") as f:
yaml.dump(args, 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"] model_name = meta_args["model_name"]
embed_dim = meta_args["embed_dim"] embed_dim = meta_args["embed_dim"]