wilson1yan / VideoGPT

MIT License
962 stars 115 forks source link

Issue with memory only with your code #25

Open Scienceseb opened 2 years ago

Scienceseb commented 2 years ago

I got the following error while running train_vqvae.py:

OSError: [WinError 1455] The paging file is too small for this operation to complete. Error loading "C:\Users\sebdb\Anaconda3\envs\pythonProject\lib\site-packages\torch\lib\cusolver64_10.dll" or one of its dependencies

This is an issue related to the memory (I have 32GB of ram with a GTX 1080TI (11GB )).

I just tried loading UFC101 : data = VideoData(args), this work but then the error appear with:

  data.train_dataloader()
  data.test_dataloader()

Those are the parameters I used:

    parser.add_argument('--sequence_length', type=int, default=8)
    parser.add_argument('--resolution', type=int, default=64)
    parser.add_argument('--batch_size', type=int, default=1)
    parser.add_argument('--num_workers', type=int, default=0) #parser.add_argument('--num_workers', type=int, default=1) also same error

Here my paging file size: 98GB... image

wilson1yan commented 2 years ago

Hi,

I'm not too familiar with Windows so I'm not too sure how to resolve this problem. However, the code does seem to work fine for me if I run it in WSL (though it is kind of hassle to update things to integrate with CUDA if you haven't already).