Open hanklin3 opened 1 year ago
Hi, What exactly did you run? which flags did you use? and - which pytorch version are you using?
Training commands:
python3 train.py --object rose
Python 3.8.5
Torch '1.13.0+cu117'
kornia '0.6.8'
4 cores cpu ( not sure if that has anything to do with the 'size 4')
requirements.txt
einops
imageio
imath
kornia>0.2.0
matplotlib
numpy
opencv_python
scikit_image
scikit-image
tensorboard
torch>1.8
torchvision>0.9
Have to fix kornia>0.2.0 cause the old version missing one argument param.
I don't know if above error has to do with the speed_up_calculation, my temp fix
if not speed_up_calculation or (speed_up_calculation and idx < len(points_to_calc)):
shadow_hat[points_to_calc[idx][0].long(), points_to_calc[idx][1].long()] = 1
continue
I was testing with
python3 train.py --object cactus --save_dir ../models/ --speed slow
but it ran too slow so I stopped it.
please refrain from using speed==slow, I left the code there only since I referred to it in the paper - but also as stated in the paper, it does not cause any increase in accuracy.
Please let me know if using speed==fast has any issues on your end.
I got the error using the default (--speed fast
).
And so I was testing if --speed slow
would work instead
I pushed a fix. Please try again and let me know if it's fixed on your end.
thank you! I will try it
Hi I'm getting this error.