快捷方式

ts.torch_handler.unit_tests package

子包

子模組

ts.torch_handler.unit_tests.test_base_handler module

BaseHandler 類的基本單元測試。確保它可以載入並執行示例模型

ts.torch_handler.unit_tests.test_base_handler.handler(base_model_context)[source]
ts.torch_handler.unit_tests.test_base_handler.test_batch_handle(handler, base_model_context)[source]
ts.torch_handler.unit_tests.test_base_handler.test_inference_with_profiler_works_with_custom_initialize_method(handler, base_model_context)[source]
ts.torch_handler.unit_tests.test_base_handler.test_single_handle(handler, base_model_context)[source]

ts.torch_handler.unit_tests.test_envelopes module

BaseHandler 類的基本單元測試。確保它可以載入並執行示例模型

ts.torch_handler.unit_tests.test_envelopes.handle_fn(base_model_context)[source]
ts.torch_handler.unit_tests.test_envelopes.test_binary(base_model_context)[source]
ts.torch_handler.unit_tests.test_envelopes.test_body(handle_fn, base_model_context)[source]
ts.torch_handler.unit_tests.test_envelopes.test_json(handle_fn, base_model_context)[source]
ts.torch_handler.unit_tests.test_envelopes.test_json_batch(handle_fn, base_model_context)[source]
ts.torch_handler.unit_tests.test_envelopes.test_json_double_batch(handle_fn, base_model_context)[source]

更復雜的測試案例。確保模型可以混合(mux)多個批次並返回解混合(demux)的結果

ts.torch_handler.unit_tests.test_image_classifier module

ImageClassifier 類的基本單元測試。確保它可以載入並執行示例模型

ts.torch_handler.unit_tests.test_image_classifier.context(model_dir, model_name)[source]
ts.torch_handler.unit_tests.test_image_classifier.handler(context)[source]
ts.torch_handler.unit_tests.test_image_classifier.image_bytes()[source]
ts.torch_handler.unit_tests.test_image_classifier.model_dir(tmp_path_factory, model_name)[source]
ts.torch_handler.unit_tests.test_image_classifier.model_name()[source]
ts.torch_handler.unit_tests.test_image_classifier.test_handle(context, image_bytes, handler)[source]
ts.torch_handler.unit_tests.test_image_classifier.test_handle_explain(context, image_bytes, handler)[source]

ts.torch_handler.unit_tests.test_image_segmenter module

ImageSegmenter 類的基本單元測試。確保它可以載入並執行示例模型

ts.torch_handler.unit_tests.test_image_segmenter.context(model_dir, model_name)[source]
ts.torch_handler.unit_tests.test_image_segmenter.handler(context)[source]
ts.torch_handler.unit_tests.test_image_segmenter.image_bytes()[source]
ts.torch_handler.unit_tests.test_image_segmenter.model_dir(tmp_path_factory, model_name)[source]
ts.torch_handler.unit_tests.test_image_segmenter.model_name()[source]
ts.torch_handler.unit_tests.test_image_segmenter.test_handle(handler, context, image_bytes)[source]

ts.torch_handler.unit_tests.test_mnist_kf module

ts.torch_handler.unit_tests.test_object_detector module

ObjectDetector 類的基本單元測試。確保它可以載入並執行示例模型

ts.torch_handler.unit_tests.test_object_detector.context(model_dir, model_name)[source]
ts.torch_handler.unit_tests.test_object_detector.handler(context)[source]
ts.torch_handler.unit_tests.test_object_detector.image_bytes()[source]
ts.torch_handler.unit_tests.test_object_detector.model_dir(tmp_path_factory, model_name)[source]
ts.torch_handler.unit_tests.test_object_detector.model_name()[source]
ts.torch_handler.unit_tests.test_object_detector.test_handle(handler, context, image_bytes)[source]

模組內容

文件

訪問全面的 PyTorch 開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源