diff --git a/README.md b/README.md index e6a964c..5b158db 100644 --- a/README.md +++ b/README.md @@ -33,6 +33,9 @@ |--------------------------------------------------|-------------------------------------------------------|---------------------------------------------|---------------------------------------------------| | | | | | +## 🤗 Community Contribution Leaderboard +1. By [@visualbruno](https://github.com/visualbruno) + - ComfyUI-Hunyuan3d-2-1: https://github.com/visualbruno/ComfyUI-Hunyuan3d-2-1 ## ☯️ **Hunyuan3D 2.1** diff --git a/hy3dshape/README-zh.md b/hy3dshape/README-zh.md deleted file mode 100644 index b526a7e..0000000 --- a/hy3dshape/README-zh.md +++ /dev/null @@ -1,47 +0,0 @@ -# Hunyuan3D-2.1-Shape - - -# 训练 - -我们会展示小数据集上DiT的训练全流程 - -## 数据预处理 - -渲染和水密化参考[链接](tools/README.md),最终得到如下结构 - -``` yaml -dataset/preprocessed/{uid} -├── geo_data -│ ├── {uid}_sdf.npz -│ ├── {uid}_surface.npz -│ └── {uid}_watertight.obj -└── render_cond - ├── 000.png - ├── ... - ├── 023.png - ├── mesh.ply - └── transforms.json -``` - -我们提供了一个8个case(均来自Objaverse-XL)预处理后的结果在 tools/mini_trainset,可以直接用于过拟合训练 - - - -## 启动训练 - -我们提供了可供参考的训练配置文件和启动脚本(默认单机8卡deepspeed训练),用户根据需要自行修改。 - -配置文件 -``` -configs/dit-from-scratch-overfitting-flowmatching-dinog518-bf16-lr1e4-1024.yaml -``` -启动脚本 - -``` -export node_num=1 -export node_rank=0 -export master_ip=0.0.0.0 # set your master_ip -export config='configs/dit-from-scratch-overfitting-flowmatching-dinog518-bf16-lr1e4-1024.yaml' -export output_dir='output_folder/dit/overfitting' -bash scripts/train_deepspeed.sh $node_num $node_rank $master_ip $config $output_dir -``` \ No newline at end of file diff --git a/hy3dshape/README.md b/hy3dshape/README.md index d7b864e..c81fc15 100644 --- a/hy3dshape/README.md +++ b/hy3dshape/README.md @@ -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 -``` \ No newline at end of file + +# 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 +``` diff --git a/hy3dshape/configs/hunyuandit-mini-overfitting-flowmatching-dinol518-bf16-lr1e4-4096.yaml b/hy3dshape/configs/hunyuandit-mini-overfitting-flowmatching-dinol518-bf16-lr1e4-4096.yaml index 62cae73..bc3ec2a 100644 --- a/hy3dshape/configs/hunyuandit-mini-overfitting-flowmatching-dinol518-bf16-lr1e4-4096.yaml +++ b/hy3dshape/configs/hunyuandit-mini-overfitting-flowmatching-dinol518-bf16-lr1e4-4096.yaml @@ -1,6 +1,6 @@ name: "HunyuanDiT flowmatching; VAE: 4096 token length; ImageEncoder: DINO-v2 Large; ImageSize: 518" -# oversitting successfully cost 68G memory under current settings -# you can adjust model arch or batch_size according to your GPU memory +# Overfitting successfully cost 68G memory under current settings +# You can adjust model arch or batch_size according to your GPU memory training: steps: 10_0000_0000