H3D One-Sample Overfit Tmux Run
Started from Cursor on GPU 2 and GPU 3.
Sample
edit_id:del_0008dc75fb3648f2af4ca8c4d711e53e_000obj_id:0008dc75fb3648f2af4ca8c4d711e53eedit_type:deletion- prompt:
Remove the detailed hand views from the character model.
Hyperparameters
max_samples:1max_steps:5000batch_size_per_gpu:1batch_split:1lr:0.0001i_print:1i_log:1i_save:100no_sample_images:true- trainable params:
--train_only_editing_weights
Configs
- Stage 1:
configs/editing/ss_flow_img_dit_L_16l8_fp16_h3d_overfit1.json - Stage 2:
configs/editing/slat_flow_img_dit_L_64l8p2_fp16_h3d_overfit1.json
Commands
Stage 1 on GPU 2:
CUDA_VISIBLE_DEVICES=2 PYTHONUNBUFFERED=1 torchrun --nproc_per_node=1 --master_port=12381 train_torchrun.py \
--config configs/editing/ss_flow_img_dit_L_16l8_fp16_h3d_overfit1.json \
--output_dir ./work_dirs/Editing_Training/h3d_overfit1_img_to_voxel_5k \
--train_only_editing_weights
Stage 2 on GPU 3:
CUDA_VISIBLE_DEVICES=3 PYTHONUNBUFFERED=1 torchrun --nproc_per_node=1 --master_port=12382 train_torchrun.py \
--config configs/editing/slat_flow_img_dit_L_64l8p2_fp16_h3d_overfit1.json \
--output_dir ./work_dirs/Editing_Training/h3d_overfit1_voxel_to_texture_5k \
--train_only_editing_weights
Logs
- Stage 1:
logs/h3d_overfit1_s1_gpu2.log - Stage 2:
logs/h3d_overfit1_s2_gpu3.log
Tmux
- session:
h3d_overfit1 - windows:
s1_gpu2,s2_gpu3