Closed pesser closed 4 years ago
Changed as suggested + added an option to delete model outputs after callbacks are finished (otherwise the amount of written data can become a bit out of control, especially when evaluating fid scores with ~10k samples)
fixed a small bug, where debug mode monkey patched the dataset size to a size larger than the dataset. Quick, review and merge or I'll keep adding to this PR ;)
support segmentation masks and fix off by one error.