Python set_soft_device_placementの例

コード例 #1

ファイル: automatic_outside_compilation_test.py プロジェクト: TheVinhLuong102/keras

 def setUp(self):
   super(AutoOutsideCompilationWithKerasTest, self).setUp()
   tf.compat.v1.enable_v2_behavior()
   set_soft_device_placement(True)
   self.summary_dir = self.get_temp_dir()

コード例 #2

 def setUp(self):
     super().setUp()
     set_soft_device_placement(True)
     self.summary_dir = self.get_temp_dir()