Issue with SegResNet network

The SegResNet network works only for the GPU and not the CPU.
Upon CPU execution following error is encountered:
ERROR: TF run failed 3 : Conv3DBackpropInputOpV2 only supports NDHWC on the CPU.
[[{{node UpscaleConvOne_0/upsample_semilinear/conv3d_transpose}}]]

Please let me know a workaround for the same.


Thanks for your interest in Clara Train SDK. Our SDK is optimized for GPUs, we don’t support training on CPU. In general training a CNN on a CPU is not recommended as it would take very long time.

Please note we have recently release clara train V4.0 based on MONAI which uses PyTorch. Please check out the notebooks to get you started clara-train-examples/PyTorch/NoteBooks at master · NVIDIA/clara-train-examples · GitHub