mirror of
https://github.com/Laurent2916/REVA-QCAV.git
synced 2024-11-09 15:02:03 +00:00
fix: dataset paths in wandb config file
This commit is contained in:
parent
96fcc64ea6
commit
c374673786
|
@ -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,
|
||||||
|
|
|
@ -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/"
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue