Epoch 0:  15%|▏| 500/3232 [10:04<55:04,  1.21s/it, loss=0.311, v_num=0, train/loss_simple_step=0.0769, train/loss_vlb_sC:\Users\Ben\miniconda3\envs\SD-Optimized\lib\site-packages\pytorch_lightning\callbacks\model_checkpoint.py:644: UserWarning: ModelCheckpoint(monitor='val/loss_simple_ema') not found in the returned metrics: ['train/loss_simple', 'train/loss_simple_step', 'train/loss_vlb', 'train/loss_vlb_step', 'train/loss', 'train/loss_step', 'global_step']. HINT: Did you call self.log('val/loss_simple_ema', value) in the LightningModule?
  warning_cache.warn(m)
Epoch 0, global step 499: val/loss_simple_ema was not in top 1

I get to 15% then this happens... python then crashes.