tensor(42380.2852, device=‘cuda:0’, grad_fn=)
tensor(124045.8906, device=‘cuda:0’, grad_fn=)
Gradient overflow.  Skipping step, loss scaler 0 reducing loss scale to 32768.0
tensor(917312., device=‘cuda:0’, grad_fn=)
Gradient overflow.  Skipping step, loss scaler 0 reducing loss scale to 16384.0
tensor(422464., device=‘cuda:0’, grad_fn=)
tensor(129528., device=‘cuda:0’, grad_fn=)
Gradient overflow.  Skipping step, loss scaler 0 reducing loss scale to 8192.0
tensor(nan, device=‘cuda:0’, grad_fn=)
Gradient overflow.  Skipping step, loss scaler 0 reducing loss scale to 4096.0
tensor(nan, device=‘cuda:0’, grad_fn=)

These are the loss values for the first couple of steps