From d2343d0d9acb77461a260450ee1c96accaadcb32 Mon Sep 17 00:00:00 2001
From: LinghaoChan <Linghao_Chan@163.com>
Date: Mon, 18 Nov 2024 09:22:59 +0800
Subject: [PATCH] update train test generate

---
 generate.sh | 6 ++++++
 test.sh     | 1 +
 train.sh    | 1 +
 3 files changed, 8 insertions(+)
 create mode 100644 generate.sh
 create mode 100644 test.sh
 create mode 100644 train.sh

diff --git a/generate.sh b/generate.sh
new file mode 100644
index 0000000..4f49aed
--- /dev/null
+++ b/generate.sh
@@ -0,0 +1,6 @@
+python -m scripts.generate --input_text ./assets/prompts.txt \
+--motion_length 8 \
+--self_attention \
+--no_eff \
+--opt_path ./checkpoints/t2m/release/opt.txt \
+--edit_mode
\ No newline at end of file
diff --git a/test.sh b/test.sh
new file mode 100644
index 0000000..042afb7
--- /dev/null
+++ b/test.sh
@@ -0,0 +1 @@
+python -m scripts.evaluation --opt_path ./checkpoints/t2m/release/opt.txt --self_attention --no_eff
diff --git a/train.sh b/train.sh
new file mode 100644
index 0000000..ae188b8
--- /dev/null
+++ b/train.sh
@@ -0,0 +1 @@
+accelerate --mixed_precision "no" launch --config_file 1gpu.yaml --gpu_ids 0 -m scripts.train --name release --model-ema --dataset_name t2m --dropout 0.1 --lr 1e-4 --no_eff --self_attention --edit_mode