Updated dataset in gdumb with replay memory
import copy
from collections import defaultdict
-from typing import Optional, Dict, Any
+from typing import Dict, Any
import numpy as np
import torch
@@ -222,6 +222,8 @@ class GDumbPlugin(StrategyPlugin):
self.counter[target_value] += 1
+ strategy.current_data = self.ext_mem
+
class EvaluationPlugin(StrategyPlugin):
"""
