Make cache saving optional (#2977)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-04-30 00:56:44 +02:00
1 parent a833ee2a46
commit 955eea8b96
1 file changed
+5 -2
+5 -2
View File
@@ -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):