LilitYolyan / CutPaste

Unofficial implementation of Google "CutPaste: Self-Supervised Learning for Anomaly Detection and Localization" in PyTorch
MIT License
114 stars 25 forks source link

Improved training loop #2

Closed AnnaManasyan closed 2 years ago

AnnaManasyan commented 2 years ago

The request contains improved version of training loop.