Multi-GPU Support
JeremyCJM opened this issue · 1 comments
JeremyCJM commented
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?
m-tassano commented
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,