Hello everyone, I did some testing for the

@pytorch_parallel

decorator, but it didn't seem to work for me when combining it with

@batch

. It did work locally, but when using

@batch

, the ec2 instances immediately terminate and the multi-node bach job gets stuck in RUNNABLE status