Skip to content

GitLab

Unverified Commit cf1a1eed authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Add missing checkpoint for doctest (#21258)

parent 5603f78f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/mobilenet_v2/modeling_mobilenet_v2.py
View file @ cf1a1eed
...@@ -637,6 +637,7 @@ class MobileNetV2ForImageClassification(MobileNetV2PreTrainedModel): ...@@ -637,6 +637,7 @@ class MobileNetV2ForImageClassification(MobileNetV2PreTrainedModel):
@add_start_docstrings_to_model_forward(MOBILENET_V2_INPUTS_DOCSTRING) @add_start_docstrings_to_model_forward(MOBILENET_V2_INPUTS_DOCSTRING)
@add_code_sample_docstrings( @add_code_sample_docstrings(
checkpoint=_CHECKPOINT_FOR_DOC,
output_type=ImageClassifierOutputWithNoAttention, output_type=ImageClassifierOutputWithNoAttention,
config_class=_CONFIG_FOR_DOC, config_class=_CONFIG_FOR_DOC,
expected_output=_IMAGE_CLASS_EXPECTED_OUTPUT, expected_output=_IMAGE_CLASS_EXPECTED_OUTPUT,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment