Questions about training about econ HOT 3 CLOSED

The training was conducted on either my local workstation with RTX 5000 or MPI cluster with A100. The 4-batch should be set for cluster training, and smaller for local workstation.

But anyway, the batch size does not affect the final performance a lot, and actually, you can train the IF-Net++ with smaller resolution (256 → 128), this won't significantly affect the final performance. And the IF-Nets++ trained with 128 resolution could be directly applied into volumetric shapes with 256^3, as it's fully convolutional.

from econ.

In my conda env, PyTorch-Lightning==2.1.0

As for the GPU, I use two Quadro RTX 5000 with 16x2 GB memory.

from econ.

In my conda env, PyTorch-Lightning==2.1.0

As for the GPU, I use two Quadro RTX 5000 with 16x2 GB memory.

That's indeed strange. The combined memory of two RTX 3090 GPUs is 2 * 24GB, and yet I could only set the batch size to 2. Did you perhaps optimize the code further in subsequent updates?

from econ.