Questions for applying on custom dataset

[Michael-Evans-Savitar]

I have met errors when executing the command: python test.py --data_root $DATA_ROOT --resize $WIDTH,$HEIGHT --crop $WIDTH,$HEIGHT --num_src $NUM_SRC. The error is : Traceback (most recent call last): File "test.py", line 105, in outputs, refined_depth, prob_maps = model(mode=args.mode) File "/mnt/nas/algorithm/anaconda3/envs/dsgs/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1190, in _call_impl return forward_call(*input, kwargs) File "/mnt/nas/algorithm/anaconda3/envs/dsgs/lib/python3.8/site-packages/torch/nn/parallel/data_parallel.py", line 171, in forward outputs = self.parallel_apply(replicas, inputs, kwargs) File "/mnt/nas/algorithm/anaconda3/envs/dsgs/lib/python3.8/site-packages/torch/nn/parallel/data_parallel.py", line 181, in parallel_apply return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)]) File "/mnt/nas/algorithm/anaconda3/envs/dsgs/lib/python3.8/site-packages/torch/nn/parallel/parallel_apply.py", line 89, in parallel_apply output.reraise() File "/mnt/nas/algorithm/anaconda3/envs/dsgs/lib/python3.8/site-packages/torch/_utils.py", line 543, in reraise raise exception TypeError: Caught TypeError in replica 0 on device 0. Original Traceback (most recent call last): File "/mnt/nas/algorithm/anaconda3/envs/dsgs/lib/python3.8/site-packages/torch/nn/parallel/parallel_apply.py", line 64, in _worker output = module(*input, *kwargs) File "/mnt/nas/algorithm/anaconda3/envs/dsgs/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1190, in _call_impl return forward_call(input, kwargs) TypeError: forward() missing 3 required positional arguments: 'sample', 'depth_nums', and 'interval_scales'