Update trainer.py

trainer.py

@@ -51,7 +51,7 @@ class Trainer:
         out_path = train_data_dir / 'target' / f'mask.jpg'
         target_mask.save(out_path, format='JPEG', quality=100)
         out_path = output_dir / f'mask.jpg'
-        target_image.save(out_path, format='JPEG', quality=100)
+        target_mask.save(out_path, format='JPEG', quality=100)
 
     def join_library_org(self) -> None:
         subprocess.run(