How to predict with model trained with multiple GPUs?

[YaoZhang93]

Hi Fabian,

I trained a nnU-Net with 2 GPUs following the instruction:

When I predict using this model by:

I got a error:

Did I get it wrong?
Do you have any suggestion to make predictions with a multi-gpu model?

Thanks,
Yao

[YaoZhang93]

The problem is located here:

The multi-gpu model accept different initial variables.
I bypass the problem by manually define these variables.

[FabianIsensee]

The multi-GPU implementation is not very good at the moment which I why i explicitly recommend not to use it ;-) This is something I need to improve in the future.