Python skip_if_cpu_is_not_supported 예제들

예제 #1

 def test_e2e_on_cpu_person(self):
     skip_if_cpu_is_not_supported(self.template_file)
     classes = 'person'
     self.do_finetuning(classes, on_gpu=False)
     self.do_evaluation(classes, on_gpu=False)
     self.do_export(classes, on_gpu=False)
     self.do_evaluation_of_exported_model(classes)

예제 #2

파일: train_tests_custom_object_detection.py 프로젝트: oahmed24/training_extensions

 def test_finetuning_with_classes_on_cpu(self):
     skip_if_cpu_is_not_supported(self.template_file)
     self.do_finetuning_with_classes(on_gpu=False)

예제 #3

파일: train_tests_custom_object_detection.py 프로젝트: oahmed24/training_extensions

 def test_e2e_on_cpu(self):
     skip_if_cpu_is_not_supported(self.template_file)
     self.do_finetuning(on_gpu=False)
     self.do_evaluation(on_gpu=False)
     self.do_export(on_gpu=False)
     self.do_evaluation_of_exported_model()