mm/citybert
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

frozen layers more epochs

Browse Source
This commit is contained in:
Michael Pilosov 2026-05-25 21:45:10 +00:00
parent 8f4d4c1057
commit 8af279771a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Makefile
View File

@ -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..."