update train test generate

IDEA-Research · Nov 18, 2024 · d2343d0 · d2343d0
1 parent 34bd8ff
commit d2343d0
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/generate.sh 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
diff --git a/test.sh 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
@@ -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
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		python -m scripts.evaluation --opt_path ./checkpoints/t2m/release/opt.txt --self_attention --no_eff
Original file line number	Diff line number	Diff line change
		@@ -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