i can do x4 in runpod, do i need to setup anything aside from the config??
i can do x4 in runpod, do i need to setup anything aside from the config??

with torch.enable_grad(), device_autocast_ctx, torch.cpu.amp.autocast(**ctx.cpu_autocast_kwargs): # type: ignore[attr-defined]
steps: 0%|▍ | 28/6500 [09:25<36:18:15, 20.19s/it, avr_loss=0.207]