Spaces:
Running
on
Zero
Running
on
Zero
| model: | |
| base_learning_rate: 5.0e-05 | |
| target: hunyuanpaintpbr.model.HunyuanPaint | |
| params: | |
| num_view: 6 | |
| view_size: 512 | |
| drop_cond_prob: 0.1 | |
| noise_in_channels: 12 | |
| stable_diffusion_config: | |
| pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1 | |
| custom_pipeline: ./hunyuanpaintpbr | |
| data: | |
| target: src.data.objaverse_hunyuan.DataModuleFromConfig | |
| params: | |
| batch_size: 1 | |
| num_workers: 4 | |
| train: | |
| - | |
| target: src.data.dataloader.objaverse_loader_forTexturePBR.TextureDataset | |
| params: | |
| num_view: 6 | |
| json_path: train_examples/examples.json | |
| validation: | |
| - | |
| target: src.data.dataloader.objaverse_loader_forTexturePBR.TextureDataset | |
| params: | |
| num_view: 6 | |
| json_path: train_examples/examples.json | |
| lightning: | |
| modelcheckpoint: | |
| params: | |
| every_n_train_steps: 10000 | |
| save_top_k: -1 | |
| save_last: true | |
| callbacks: {} | |
| trainer: | |
| benchmark: true | |
| max_epochs: -1 | |
| gradient_clip_val: 1.0 | |
| val_check_interval: 1000 | |
| num_sanity_val_steps: 0 | |
| accumulate_grad_batches: 1 | |
| check_val_every_n_epoch: null # if not set this, validation does not run | |
| init_control_from: | |
| resume_from: | |