Reset EWC to less verbose logging
@@ -4,8 +4,7 @@ from avalanche.benchmarks import PermutedMNIST, SplitMNIST
from avalanche.training.strategies import EWC
from avalanche.models import SimpleMLP
from avalanche.evaluation.metrics import ExperienceForgetting, \
- accuracy_metrics, loss_metrics, cpu_usage_metrics, timing_metrics, \
- gpu_usage_metrics, ram_usage_metrics
+ accuracy_metrics, loss_metrics
from avalanche.logging import InteractiveLogger, TextLogger
from avalanche.training.plugins import EvaluationPlugin
@@ -51,26 +50,10 @@ def main(args):
eval_plugin = EvaluationPlugin(
accuracy_metrics(
minibatch=True, epoch=True, experience=True, stream=True),
- loss_metrics(minibatch=True, epoch=True, experience=True, stream=True),
- cpu_usage_metrics(
- minibatch=False, epoch=True, experience=True, stream=True),
- #timing_metrics(
- # minibatch=True, epoch=True, experience=True, stream=True),
- ram_usage_metrics(
- minibatch=False, epoch=True, experience=True, stream=True),
- gpu_usage_metrics(
- args.cuda, minibatch=False, epoch=True, experience=True, stream=True),
+ loss_metrics(
+ minibatch=True, epoch=True, experience=True, stream=True),
ExperienceForgetting(),
- loggers=[interactive_logger, text_logger])
-
- # eval_plugin = EvaluationPlugin(
- # accuracy_metrics(
- # minibatch=True, epoch=True, experience=True, stream=True),
- # loss_metrics(minibatch=True, epoch=True, experience=True, stream=True),
- # ExperienceForgetting(),
- # ram_usage_metrics(experience=True, epoch=True, stream=True),
- # gpu_usage_metrics(args.cuda, every=0.5, experience=True, epoch=True, stream=True),
- # loggers=[interactive_logger])
+ loggers=[interactive_logger])
# create strategy
strategy = EWC(model, optimizer, criterion, args.ewc_lambda,
