MEDIAR / predict.sh
ghlee94's picture
Init
2a13495
raw
history blame
129 Bytes
python predict.py -i "./inputs" -o "./outputs" --device "cuda:0" --model_path="./main_model.pt" --model_path2="./sub_model.pth"