m-tassano/fastdvdnet

Multi-GPU Support

JeremyCJM opened this issue · 1 comments

It seems that during validation, only one patch is loaded (i.e. the test batch size is 1), which limits the code to be run on multiple GPUs. Could you provide the multi-GPU version of the code so that the training and testing can be scaled up?

Hi,
Unfortunately, I won't be able to provide this code, as at the moment I don't have access to a multi-GPU machine on which to test this feature. In any case, you are welcome to propose a PR with this functionality. Best,