Added stream forgetting to the example
@@ -29,7 +29,7 @@ from torchvision.transforms import ToTensor, RandomCrop
from avalanche.benchmarks import nc_scenario
from avalanche.logging import InteractiveLogger, WandBLogger
from avalanche.training.plugins import EvaluationPlugin
-from avalanche.evaluation.metrics import ExperienceForgetting, StreamConfusionMatrix, \
+from avalanche.evaluation.metrics import forgetting_metrics, StreamConfusionMatrix, \
accuracy_metrics, loss_metrics, cpu_usage_metrics, timing_metrics, \
gpu_usage_metrics, ram_usage_metrics, disk_usage_metrics, MAC_metrics
from avalanche.models import SimpleMLP
@@ -75,7 +75,7 @@ def main(args):
minibatch=True, epoch=True, epoch_running=True, experience=True, stream=True),
loss_metrics(
minibatch=True, epoch=True, epoch_running=True, experience=True, stream=True),
- ExperienceForgetting(),
+ forgetting_metrics(experience=True, stream=True),
StreamConfusionMatrix(),
cpu_usage_metrics(
minibatch=True, epoch=True, experience=True, stream=True),
