KawasakiAkasei/Hunyuan3D_2.1_Low_VRAM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update readme

Browse Source
This commit is contained in:
Huiwenshi
2025-06-26 16:34:51 +08:00
parent 7c92655a0d
commit e59169a8ec
4 changed files with 16 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
hy3dshape/README.md
View File

@@ -45,10 +45,17 @@ configs/hunyuandit-mini-overfitting-flowmatching-dinog518-bf16-lr1e4-512.yaml
Launch Script
```
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
export num_gpu_per_node=8
export node_num=1
export node_rank=0
export master_ip=0.0.0.0 # set your master_ip
export config=configs/hunyuandit-mini-overfitting-flowmatching-dinog518-bf16-lr1e4-512.yaml
export output_dir=output_folder/dit/overfitting
bash scripts/train_deepspeed.sh $node_num $node_rank $master_ip $config $output_dir
```
# export config=configs/hunyuandit-finetuning-flowmatching-dinol518-bf16-lr1e5-4096.yaml
# export output_dir=output_folder/dit/fintuning_lr1e5
export config=configs/hunyuandit-mini-overfitting-flowmatching-dinol518-bf16-lr1e4-4096.yaml
export output_dir=output_folder/dit/overfitting_depth_16_token_4096_lr1e4
bash scripts/train_deepspeed.sh $node_num $node_rank $num_gpu_per_node $master_ip $config $output_dir
```