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.
48 lines
1.2 KiB
48 lines
1.2 KiB
5 months ago
|
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
|