コード例 #1
0
ファイル: test_plugins.py プロジェクト: bbeatrix/avalanche 
 def test_replay_balanced_memory(self):
     mem_size = 25
     policies = [None,
                 ExperienceBalancedStoragePolicy({}, mem_size=mem_size),
                 ClassBalancedStoragePolicy({}, mem_size=mem_size)]
     for policy in policies:
         self._test_replay_balanced_memory(policy, mem_size)
コード例 #2
0
ファイル: test_replay.py プロジェクト: stjordanis/avalanche 
    def test_balancing(self):
        p1 = ExperienceBalancedStoragePolicy({}, 100, adaptive_size=True)
        p2 = ClassBalancedStoragePolicy({}, 100, adaptive_size=True)

        for policy in [p1, p2]:
            self.assert_balancing(policy)