Python get_auxiliary_parametersの例

コード例 #1

ファイル: optimizer_test.py プロジェクト: zacker150/pytorch

 def check_optimizer(self, optimizer):
     self.assertTrue(optimizer.get_auxiliary_parameters().shared)
     self.assertTrue(optimizer.get_auxiliary_parameters().local)
     self.assertTrue(workspace.HasBlob("optimizer_iteration"))
     iteration_tensor = workspace.FetchBlob("optimizer_iteration")
     np.testing.assert_allclose(np.array([2000]),
                                iteration_tensor,
                                atol=1e-5)
     for param in optimizer.get_auxiliary_parameters().shared:
         workspace.FetchBlob(param)
     for param in optimizer.get_auxiliary_parameters().local:
         workspace.FetchBlob(param)

コード例 #2

ファイル: optimizer_test.py プロジェクト: Sissipei/caffe2

 def check_optimizer(self, optimizer):
     self.assertTrue(optimizer.get_auxiliary_parameters().shared)
     self.assertTrue(optimizer.get_auxiliary_parameters().local)
     self.assertTrue(workspace.HasBlob("optimizer_iteration"))
     iteration_tensor = workspace.FetchBlob("optimizer_iteration")
     np.testing.assert_allclose(np.array([2000]),
                                iteration_tensor,
                                atol=1e-5)
     for param in optimizer.get_auxiliary_parameters().shared:
         workspace.FetchBlob(param)
     for param in optimizer.get_auxiliary_parameters().local:
         workspace.FetchBlob(param)

コード例 #3

ファイル: optimizer_test.py プロジェクト: aimsky/caffe2

 def check_optimizer(self, optimizer):
     self.assertFalse(optimizer.get_auxiliary_parameters().shared)
     self.assertTrue(optimizer.get_auxiliary_parameters().local)
     for param in optimizer.get_auxiliary_parameters().local:
         workspace.FetchBlob(param)

コード例 #4

ファイル: optimizer_test.py プロジェクト: aimsky/caffe2

 def check_optimizer(self, optimizer):
     self.assertTrue(optimizer.get_auxiliary_parameters().shared)
     self.assertFalse(optimizer.get_auxiliary_parameters().local)
     for param in optimizer.get_auxiliary_parameters().shared:
         tensor = workspace.FetchBlob(param)
         np.testing.assert_allclose(np.array([1.0]), tensor, atol=1e-5)

コード例 #5

ファイル: optimizer_test.py プロジェクト: Sissipei/caffe2

 def check_optimizer(self, optimizer):
     self.assertFalse(optimizer.get_auxiliary_parameters().shared)
     self.assertTrue(optimizer.get_auxiliary_parameters().local)
     for param in optimizer.get_auxiliary_parameters().local:
         workspace.FetchBlob(param)

コード例 #6

ファイル: optimizer_test.py プロジェクト: Sissipei/caffe2

 def check_optimizer(self, optimizer):
     self.assertTrue(optimizer.get_auxiliary_parameters().shared)
     self.assertFalse(optimizer.get_auxiliary_parameters().local)
     for param in optimizer.get_auxiliary_parameters().shared:
         tensor = workspace.FetchBlob(param)
         np.testing.assert_allclose(np.array([1.0]), tensor, atol=1e-5)