"docs/vscode:/vscode.git/clone" did not exist on "8cf4a6f0a63ed3aeed68192a9304fed2bd0ce100"
Unverified Commit 6f25d107 authored by S.Kishore's avatar S.Kishore Committed by GitHub
Browse files

Added type hints to ResNetForImageClassification (#19084)

* Added type hints to ResNetForImageClassification

* Resolved check_repository_consistency failure issue

Running fix-copies changed the type hints for RegNetForImageClassification in modeling_regnet.py file
parent fe5e7cea
...@@ -407,10 +407,10 @@ class RegNetForImageClassification(RegNetPreTrainedModel): ...@@ -407,10 +407,10 @@ class RegNetForImageClassification(RegNetPreTrainedModel):
) )
def forward( def forward(
self, self,
pixel_values: Tensor = None, pixel_values: Optional[torch.FloatTensor] = None,
labels: Tensor = None, labels: Optional[torch.LongTensor] = None,
output_hidden_states: bool = None, output_hidden_states: Optional[bool] = None,
return_dict: bool = None, return_dict: Optional[bool] = None,
) -> ImageClassifierOutputWithNoAttention: ) -> ImageClassifierOutputWithNoAttention:
r""" r"""
labels (`torch.LongTensor` of shape `(batch_size,)`, *optional*): labels (`torch.LongTensor` of shape `(batch_size,)`, *optional*):
......
...@@ -370,10 +370,10 @@ class ResNetForImageClassification(ResNetPreTrainedModel): ...@@ -370,10 +370,10 @@ class ResNetForImageClassification(ResNetPreTrainedModel):
) )
def forward( def forward(
self, self,
pixel_values: Tensor = None, pixel_values: Optional[torch.FloatTensor] = None,
labels: Tensor = None, labels: Optional[torch.LongTensor] = None,
output_hidden_states: bool = None, output_hidden_states: Optional[bool] = None,
return_dict: bool = None, return_dict: Optional[bool] = None,
) -> ImageClassifierOutputWithNoAttention: ) -> ImageClassifierOutputWithNoAttention:
r""" r"""
labels (`torch.LongTensor` of shape `(batch_size,)`, *optional*): labels (`torch.LongTensor` of shape `(batch_size,)`, *optional*):
......
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