You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

47 lines
1.2 KiB

model:
base_learning_rate: 1.0e-05
target: zero123plus.model.MVDiffusion
params:
drop_cond_prob: 0.1
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: 6
num_workers: 8
train:
target: src.data.objaverse_zero123plus.ObjaverseData
params:
root_dir: data/objaverse
meta_fname: lvis-annotations.json
image_dir: rendering_zero123plus
validation: false
validation:
target: src.data.objaverse_zero123plus.ObjaverseData
params:
root_dir: data/objaverse
meta_fname: lvis-annotations.json
image_dir: rendering_zero123plus
validation: true
lightning:
modelcheckpoint:
params:
every_n_train_steps: 1000
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