wilson1yan/VideoGPT

AssertionError: Must set max_steps argument

Closed this issue · 2 comments

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.

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

Thanks!