|
|
|
|
|
conda activate /mnt/pfs/users/chaojun.ni/wangweijie_mnt/wangweijie/miniconda3/envs/depthsplat |
|
|
CUDA_VISIBLE_DEVICES=5 python -m src.main +experiment=dl3dv \ |
|
|
data_loader.train.batch_size=1 \ |
|
|
'dataset.roots'='["datasets/dl3dv"]' \ |
|
|
dataset.view_sampler.num_target_views=8 \ |
|
|
dataset.view_sampler.num_context_views=6 \ |
|
|
dataset.min_views=1 \ |
|
|
dataset.max_views=6 \ |
|
|
trainer.max_steps=100000 \ |
|
|
trainer.num_nodes=1 \ |
|
|
model.encoder.num_scales=2 \ |
|
|
model.encoder.upsample_factor=4 \ |
|
|
model.encoder.lowest_feature_resolution=8 \ |
|
|
model.encoder.monodepth_vit_type=vitb \ |
|
|
mode=test \ |
|
|
test.save_video=false \ |
|
|
test.save_depth_concat_img=True \ |
|
|
test.save_image=True \ |
|
|
test.save_gt_image=True \ |
|
|
checkpointing.pretrained_model=checkpoints/dl3dv-256x448-volsplat_3dunet-randview2-6/checkpoints/epoch_9-step_100000.ckpt \ |
|
|
model.encoder.supervise_intermediate_depth=false \ |
|
|
output_dir=checkpoints/dl3dv-256x448-depthsplat-base-randview2-6 \ |
|
|
train.no_viz_video=True |
|
|
|
|
|
|
|
|
CUDA_VISIBLE_DEVICES=0 python -m src.main +experiment=re10k \ |
|
|
data_loader.train.batch_size=1 \ |
|
|
'dataset.roots'='["/home/aiscuser/yeqing/dataset/re10k"]' \ |
|
|
dataset.test_chunk_interval=10 \ |
|
|
trainer.max_steps=150000 \ |
|
|
model.encoder.num_scales=2 \ |
|
|
model.encoder.upsample_factor=2 \ |
|
|
model.encoder.lowest_feature_resolution=4 \ |
|
|
model.encoder.monodepth_vit_type=vitb \ |
|
|
mode=test \ |
|
|
test.save_video=false \ |
|
|
test.save_depth_concat_img=false \ |
|
|
test.save_image=false \ |
|
|
test.save_gt_image=false \ |
|
|
checkpointing.pretrained_model=/home/aiscuser/yeqing/Vol-Splat/checkpoints/re10k-256x256-unet_volsplat/checkpoints/epoch_2-step_150000.ckpt |
