From 955eea8b96397b1f1f7666c4ee42d1c6411d0755 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 30 Apr 2021 00:56:44 +0200 Subject: [PATCH] Make cache saving optional (#2977) --- utils/datasets.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/utils/datasets.py b/utils/datasets.py index 3fcdddd..43e4e59 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -502,8 +502,11 @@ class LoadImagesAndLabels(Dataset): # for training/testing x['hash'] = get_hash(self.label_files + self.img_files) x['results'] = nf, nm, ne, nc, i + 1 x['version'] = 0.1 # cache version - torch.save(x, path) # save for next time - logging.info(f'{prefix}New cache created: {path}') + try: + torch.save(x, path) # save for next time + logging.info(f'{prefix}New cache created: {path}') + except Exception as e: + logging.info(f'{prefix}WARNING: Cache directory {path.parent} is not writeable: {e}') # path not writeable return x def __len__(self):