From 8af279771a0905ce1ce03b7efa4b877eeb7e4739 Mon Sep 17 00:00:00 2001 From: Michael Pilosov Date: Mon, 25 May 2026 21:45:10 +0000 Subject: [PATCH] frozen layers more epochs --- Makefile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Makefile b/Makefile index c0f51c7..6fddb7a 100644 --- a/Makefile +++ b/Makefile @@ -8,7 +8,7 @@ data: training.csv train: train.py training.csv @echo "Training coordinate regressor..." - @bash -c 'source .venv/bin/activate && python train.py' + @bash -c 'source .venv/bin/activate && python train.py --epochs 50' eval: eval.py training.csv @echo "Evaluating 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' + @bash -c 'source .venv/bin/activate && python train.py --output-path output_frozen_layers --freeze-transformer-layers 4 --epochs 50' eval_frozen_layers: eval.py training.csv @echo "Evaluating frozen-layer coordinate regressor..."