Open Crazy-wu-20 opened 3 weeks ago
When evaluating aesthetic_quality for my own videos, I always get the error 'CUDA out of memory', which leading me not be able to batch test. Is there any problem? And according to my understanding, the model should only load one video at a time.
Example: python evaluate.py --dimension aesthetic_quality --videos_path ../test --mode=custom_input
python evaluate.py --dimension aesthetic_quality --videos_path ../test --mode=custom_input
The first video can be OK. However, for the third one, torch.cuda.OutOfMemoryError: CUDA out of memory would occur.
torch.cuda.OutOfMemoryError: CUDA out of memory
When evaluating aesthetic_quality for my own videos, I always get the error 'CUDA out of memory', which leading me not be able to batch test. Is there any problem? And according to my understanding, the model should only load one video at a time.
Example:
python evaluate.py --dimension aesthetic_quality --videos_path ../test --mode=custom_inputThe first video can be OK. However, for the third one,
torch.cuda.OutOfMemoryError: CUDA out of memorywould occur.