Lower number of exp for testing
@@ -56,7 +56,7 @@ def main(args):
# ---------
# --- SCENARIO CREATION
- scenario = SplitMNIST(n_experiences=10, seed=1234)
+ scenario = SplitMNIST(n_experiences=3, seed=1234)
# ---------
# MODEL CREATION
@@ -77,7 +77,7 @@ def main(args):
# TRAINING LOOP
print("Starting experiment...")
- f, axarr = plt.subplots(len(scenario.train_stream), 10)
+ f, axarr = plt.subplots(scenario.n_experiences, 10)
for experience in scenario.train_stream:
k = 0
print("Start of experience ", experience.current_experience)
