2D Classification Memory Allocation - Unresponsive Job

Hi @zjr, does your issue look similar to this:

Kernel panic - linux box with 2 GPUS