Training a TRT_Pose Model

I have a NVIDIA Jetson Xavier NX on which I’ve installed TRT_Pose. I’m able to run an evaluation, but I’m not able to finish a training session before the process is killed, presumably due to RAM overload. I would like to know if there are some alternative environments where I could install TRT_Pose and train a model. I tried using Amazon’s G4 Nvidia AMI instance, but it didn’t appear compatible.

This looks like a Jetson issue. Please refer to the below samlples in case useful.

For any further assistance, we recommend you to raise it to the respective platform from the below link