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
refer to the PyTorch documentation for all matter related to general usage and behavior.
.. _`Supervised Multimodal Bitransformers for Classifying Images and Text`:
https://www.github.com/salesforce/ctrl
https://github.com/facebookresearch/mmbt
.. _`torch.nn.Module`:
https://pytorch.org/docs/stable/nn.html#module
......@@ -365,4 +365,4 @@ class MMBTForClassification(nn.Module):
loss = loss_fct(logits.view(-1, self.num_labels), labels.view(-1))
outputs = (loss,) + outputs
return outputs # (loss), logits, (hidden_states), (attentions)
\ No newline at end of file
return outputs # (loss), logits, (hidden_states), (attentions)
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