Can I see the config.yaml you used here? I am trying to run the same exact code with a similar setup, but something goes wrong and I can’e even get the training to start. See more here: Cuda Error: invalid device ordinal during training on GCP cluster