3Deditformer / docs /H3D_OVERFIT_TMUX_RUN.md
zhaxie's picture
Upload 3DEditFormer source code
2274e38 verified
|
Raw
History Blame Contribute Delete
1.54 kB

H3D One-Sample Overfit Tmux Run

Started from Cursor on GPU 2 and GPU 3.

Sample

  • edit_id: del_0008dc75fb3648f2af4ca8c4d711e53e_000
  • obj_id: 0008dc75fb3648f2af4ca8c4d711e53e
  • edit_type: deletion
  • prompt: Remove the detailed hand views from the character model.

Hyperparameters

  • max_samples: 1
  • max_steps: 5000
  • batch_size_per_gpu: 1
  • batch_split: 1
  • lr: 0.0001
  • i_print: 1
  • i_log: 1
  • i_save: 100
  • no_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