From cd9da608931ff47d98ce9cff6884c06493bf020c Mon Sep 17 00:00:00 2001 From: Michael Pilosov Date: Mon, 25 May 2026 22:51:43 +0000 Subject: [PATCH] 10 epochs --- Makefile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Makefile b/Makefile index 6fddb7a..4b52ed8 100644 --- a/Makefile +++ b/Makefile @@ -16,7 +16,7 @@ eval: eval.py training.csv train_frozen_encoder: train.py training.csv @echo "Training coordinate head with frozen encoder..." - @bash -c 'source .venv/bin/activate && python train.py --output-path output_frozen_encoder --freeze-encoder --epochs 50' + @bash -c 'source .venv/bin/activate && python train.py --output-path output_frozen_encoder --freeze-encoder --epochs 10' eval_frozen_encoder: eval.py training.csv @echo "Evaluating frozen-encoder coordinate regressor..." @@ -24,7 +24,7 @@ eval_frozen_encoder: eval.py training.csv train_frozen_layers: train.py training.csv @echo "Training coordinate regressor with first transformer layers frozen..." - @bash -c 'source .venv/bin/activate && python train.py --output-path output_frozen_layers --freeze-transformer-layers 4 --epochs 50' + @bash -c 'source .venv/bin/activate && python train.py --output-path output_frozen_layers --freeze-transformer-layers 4 --epochs 10' eval_frozen_layers: eval.py training.csv @echo "Evaluating frozen-layer coordinate regressor..."