Unverified Commit b668a740 authored by Daniil Larionov's avatar Daniil Larionov Committed by GitHub
Browse files

Fixing incorrect link in model docstring

The docstring contains a link to Salesforce/CTRL repo, while the model itself is Facebookresearch/mmbt. It may be the wrong copy\paste.
parent 645713e2
...@@ -82,7 +82,7 @@ MMBT_START_DOCSTRING = r""" MMBT model was proposed in ...@@ -82,7 +82,7 @@ MMBT_START_DOCSTRING = r""" MMBT model was proposed in
refer to the PyTorch documentation for all matter related to general usage and behavior. refer to the PyTorch documentation for all matter related to general usage and behavior.
.. _`Supervised Multimodal Bitransformers for Classifying Images and Text`: .. _`Supervised Multimodal Bitransformers for Classifying Images and Text`:
https://www.github.com/salesforce/ctrl https://github.com/facebookresearch/mmbt
.. _`torch.nn.Module`: .. _`torch.nn.Module`:
https://pytorch.org/docs/stable/nn.html#module https://pytorch.org/docs/stable/nn.html#module
...@@ -365,4 +365,4 @@ class MMBTForClassification(nn.Module): ...@@ -365,4 +365,4 @@ class MMBTForClassification(nn.Module):
loss = loss_fct(logits.view(-1, self.num_labels), labels.view(-1)) loss = loss_fct(logits.view(-1, self.num_labels), labels.view(-1))
outputs = (loss,) + outputs outputs = (loss,) + outputs
return outputs # (loss), logits, (hidden_states), (attentions) return outputs # (loss), logits, (hidden_states), (attentions)
\ No newline at end of file
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