Skip to content

GitLab

Commit 8d0fb2a5 authored by mashun1's avatar mashun1
Browse files

Update train.sh

parent fa8e215a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
run/scripts/train.sh
View file @ 8d0fb2a5
...@@ -6,7 +6,7 @@ train_data_record_path="/home/vtryon/catvton_v0/datasets/train_data.jsonl" ...@@ -6,7 +6,7 @@ train_data_record_path="/home/vtryon/catvton_v0/datasets/train_data.jsonl"
eval_data_record_path="/home/vtryon/catvton_v0/datasets/eval_data.jsonl" eval_data_record_path="/home/vtryon/catvton_v0/datasets/eval_data.jsonl"
height=512 height=512
width=384 width=384
vae_subfolder="sd-vae-ft-mse" vae_subfolder="vae"
weight_dtype="bf16" weight_dtype="bf16"
max_steps=32000 max_steps=32000
logging_steps=1000 logging_steps=1000
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment