fix: dataset paths in wandb config file

This commit is contained in:
Laurent Fainsin 2022-09-13 15:38:13 +02:00
parent 96fcc64ea6
commit c374673786
2 changed files with 8 additions and 6 deletions

View file

@ -46,7 +46,8 @@ class Spheres(pl.LightningDataModule):
), ),
) )
dataset = LabeledDataset(image_dir="/dev/shm/TRAIN/", transforms=transforms) # dataset = LabeledDataset(image_dir="/dev/shm/TRAIN/", transforms=transforms)
dataset = LabeledDataset(image_dir=wandb.config.DIR_TRAIN_IMG, transforms=transforms)
# dataset = Subset(dataset, range(6 * 200)) # subset for debugging purpose # dataset = Subset(dataset, range(6 * 200)) # subset for debugging purpose
# dataset = Subset(dataset, [0] * 320) # overfit test # dataset = Subset(dataset, [0] * 320) # overfit test
@ -84,7 +85,8 @@ class Spheres(pl.LightningDataModule):
), ),
) )
dataset = RealDataset(root="/dev/shm/TEST/", transforms=transforms) # dataset = RealDataset(root="/dev/shm/TEST/", transforms=transforms)
dataset = RealDataset(root=wandb.config.DIR_VALID_IMG, transforms=transforms)
return DataLoader( return DataLoader(
dataset, dataset,

View file

@ -1,7 +1,7 @@
# DIR_TRAIN_IMG: DIR_TRAIN_IMG:
# value: "/media/disk1/lfainsin/BACKGROUND/" value: "/media/disk1/lfainsin/TRAIN_prerender/"
# DIR_VALID_IMG: DIR_VALID_IMG:
# value: "/media/disk1/lfainsin/TEST_batched/" value: "/media/disk1/lfainsin/TEST_tmp_mrcnn/"
# DIR_SPHERE: # DIR_SPHERE:
# value: "/media/disk1/lfainsin/SPHERES/" # value: "/media/disk1/lfainsin/SPHERES/"