Update train.slurm

v0lta · web-flow · commit c4937b34cd49 · 2025-03-04T17:54:22.000+01:00
diff --git a/scripts/train.slurm b/scripts/train.slurm
@@ -16,4 +16,4 @@ module load PyTorch
 
 pip install -r requirements.txt
 
-python ./src/train.py
+python ./src/train.py --data_dir data/shakespeare --model Attention

Original file line number	Diff line number	Diff line change
`@@ -16,4 +16,4 @@ module load PyTorch`
`16`	`16`
`17`	`17`	`pip install -r requirements.txt`
`18`	`18`
`19`		`-python ./src/train.py`
	`19`	`+python ./src/train.py --data_dir data/shakespeare --model Attention`