Add save_dir arg to test.test, use arg as location for saving batch jpgs

train.py

@@ -303,7 +303,8 @@ def train(hyp):
                                              model=ema.ema,
                                              single_cls=opt.single_cls,
                                              dataloader=testloader,
-                                             fast=epoch < epochs / 2)
+                                             fast=epoch < epochs / 2
+                                             save_dir=log_dir)
 
         # Write
         with open(results_file, 'a') as f: