Hi @ptrblck,
Unfortunately no,  I don’t think so b/c in train1.py the devices are determined programmatically in the pytorch lightning pl.LightningModule by calling self.device.
Any other ideas why this error might be occuring?
Thanks!