From 7933564c6637d13e8fdb6e67073f7cfc398b6207 Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Wed, 12 Oct 2022 17:33:54 +0100
Subject: [PATCH] typo

---
 aimodel/slurm-train.job | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aimodel/slurm-train.job b/aimodel/slurm-train.job
index f89e769..86368a7 100755
--- a/aimodel/slurm-train.job
+++ b/aimodel/slurm-train.job
@@ -63,7 +63,7 @@ echo ">>> Installing requirements";
 conda run -n py38 pip install -r requirements.txt;
 echo ">>> Training model";
 #shellcheck disable=SC2086
-/usr/bin/env time -v conda run -n py38 src/index.py pretrain --input "${INPUT}" --output "${dir_output}" ${ARGS};
+/usr/bin/env time -v conda run -n py38 src/index.py train --input "${INPUT}" --output "${dir_output}" ${ARGS};
 src/index.py train --input "${INPUT}" -o "${dir_output}"
 # Recommended extra args: --water-size 94 --batch-size 48 --arch convnext_i_xtiny