Spaces:
Runtime error
Runtime error
File size: 442 Bytes
660acc7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
#!/usr/bin/env bash
set -e
set -x
# Use specified checkpoint path, otherwise, default value
ckpt=${1:-"prs-eth/marigold-v1-0"}
subfolder=${2:-"eval"}
python infer.py \
--checkpoint $ckpt \
--seed 1234 \
--base_data_dir $BASE_DATA_DIR \
--denoise_steps 50 \
--ensemble_size 10 \
--processing_res 0 \
--dataset_config config/dataset/data_nyu_test.yaml \
--output_dir output/${subfolder}/nyu_test/prediction \
|