Open cs19469 opened 1 year ago
Thanks for your interest in our work, the training models of HVS-5M on all six datasets are available under the "models" folder.
Thanks a lot this helps.
Have you ever encountered such a problem? terminate called after throwing an instance of 'std::bad_alloc' what(): std::bad_alloc Aborted (core dumped) when I run the test_demo.py. and I think the GPU with 3090 is ok.
Sorry, I haven't encountered such a problem, but 3090 is certainly enough to run test_demo.py.
Hello, I am very interested in your work. I wonder if it is possible for you to provide training models on different datasets.