wilson1yan / VideoGPT

MIT License
962 stars 115 forks source link

AssertionError: Must set max_steps argument #19

Closed slerman12 closed 3 years ago

slerman12 commented 3 years ago

I am getting this error when I try to run the following:

python3 train_videogpt.py --vqvae ucf101_stride4x4x4 --data_path ./datasets/ucf101/ --gpus 4

According to the docs, max steps is disabled by default? I'm confused, do I need to pass in a value for it? What would be a good value for achieving convergence on the model?

Thanks. Looking forward to getting this model up and running.

wilson1yan commented 3 years ago

Yes, you should pass a value in for it. I believe I ran it with --max_steps 200000

slerman12 commented 3 years ago

Thanks!