fixed last syntax error
#!/usr/bin/env python3
import copy
-from typing import Optional, Sequence
+from typing import List, Optional, Sequence
import numpy as np
import torch
@@ -9,6 +9,7 @@ from torch.nn import CrossEntropyLoss, Module
from torch.optim import Optimizer
from avalanche.benchmarks.utils import concat_datasets
+from avalanche.core import SupervisedPlugin
from avalanche.models.utils import avalanche_forward
from avalanche.training import ACECriterion
from avalanche.training.plugins.evaluation import default_evaluator
@@ -54,7 +55,7 @@ class OnlineER_ACE(OnlineSupervisedTemplate):
train_passes: int = 1,
eval_mb_size: Optional[int] = 1,
device="cpu",
- plugins: Optional[Sequence["BaseSGDPlugin"]] = None,
+ plugins: Optional[List[SupervisedPlugin]] = None,
evaluator=default_evaluator(),
eval_every=-1,
peval_mode="experience",
@@ -210,7 +211,7 @@ class ER_ACE(SupervisedTemplate):
train_epochs: int = 1,
eval_mb_size: Optional[int] = 1,
device="cpu",
- plugins: Optional[Sequence["BaseSGDPlugin"]] = None,
+ plugins: Optional[List[SupervisedPlugin]] = None,
evaluator=default_evaluator(),
eval_every=-1,
peval_mode="epoch",
