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)