add checkpoint

Author: tsuba-neko

train.py

@@ -21,6 +21,7 @@
 import random
 import sys
 import time
+from pprint import pprint
 from copy import deepcopy
 from datetime import datetime
 from pathlib import Path
@@ -385,6 +386,9 @@ def train(hyp, opt, device, callbacks):  # hyp is path/to/hyp.yaml or hyp dictio
                     torch.save(ckpt, best)
                 if opt.save_period > 0 and epoch % opt.save_period == 0:
                     LOGGER.info(f'Saving checkpoint at {epoch} epochs')
+                    pprint(ckpt)
+                    pprint(opt.save_dir)
+                    pprint(opt)
                     torch.save(ckpt, w / f'epoch{epoch}.pt')
                 del ckpt
                 callbacks.run('on_model_save', last, epoch, final_epoch, best_fitness, fi)