Skip to content

Commit 4fe02fc

Browse files
committed
Add cifar100 to scenario
1 parent 76ef62b commit 4fe02fc

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

mplc/scenario.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,8 @@ def __init__(
138138
self.dataset = dataset_module.Esc50()
139139
elif dataset_name == constants.IMDB:
140140
self.dataset = dataset_module.Imdb()
141+
elif dataset_name == constants.CIFAR100:
142+
self.dataset = dataset_module.Cifar100()
141143
else:
142144
raise Exception(
143145
f"Dataset named '{dataset_name}' is not supported (yet). You can construct your own "

0 commit comments

Comments
 (0)