Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,458 Bytes
b8fee6a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
model:
base_learning_rate: 1.0e-05
target: zero123plus.model.MVDiffusionRefinement
params:
drop_cond_prob: 0.1
refinement: true
stable_diffusion_config:
pretrained_model_name_or_path: sudo-ai/zero123plus-v1.2
custom_pipeline: ./zero123plus
data:
target: src.data.objaverse_zero123plus.DataModuleFromConfig
params:
batch_size: 3
num_workers: 1
train:
target: src.data.objaverse_zero123plus.RefinementData
params:
root_dir: refinement_dataset/
gt_subpath: gt
pred_subpath: shap_e
validation: false
caption_path: captions.json
split_path: dataset_splits_fixed.json
single_view: false
single_light: true
validation:
target: src.data.objaverse_zero123plus.RefinementData
params:
root_dir: refinement_dataset/
gt_subpath: gt
pred_subpath: shap_e
validation: true
caption_path: captions.json
split_path: dataset_splits_fixed.json
single_view: false
single_light: true
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: 10000000
num_sanity_val_steps: 0
accumulate_grad_batches: 1
check_val_every_n_epoch: null # if not set this, validation does not run
|