diff --git a/README.md b/README.md index c18da5f..146bf91 100644 --- a/README.md +++ b/README.md @@ -48,6 +48,10 @@ logits = outputs.logits # shape (batch_size, num_labels, height/4, width/4) For more code examples, we refer to the [documentation](https://huggingface.co/transformers/model_doc/segformer.html#). +### License + +The license for this model can be found [here](https://github.com/NVlabs/SegFormer/blob/master/LICENSE). + ### BibTeX entry and citation info ```bibtex