diff --git a/mmseg/models/backbones/mix_transformer.py b/mmseg/models/backbones/mix_transformer.py
index aab4bcc..e4b7ea2 100644
--- a/mmseg/models/backbones/mix_transformer.py
+++ b/mmseg/models/backbones/mix_transformer.py
@@ -1,3 +1,8 @@
+# ---------------------------------------------------------------
+# Copyright (c) 2021, NVIDIA Corporation. All rights reserved.
+#
+# This work is licensed under the NVIDIA Source Code License
+# ---------------------------------------------------------------
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
diff --git a/mmseg/models/decode_heads/segformer_head.py b/mmseg/models/decode_heads/segformer_head.py
index 97d2ab6..8ada6d7 100644
--- a/mmseg/models/decode_heads/segformer_head.py
+++ b/mmseg/models/decode_heads/segformer_head.py
@@ -1,3 +1,8 @@
+# ---------------------------------------------------------------
+# Copyright (c) 2021, NVIDIA Corporation. All rights reserved.
+#
+# This work is licensed under the NVIDIA Source Code License
+# ---------------------------------------------------------------
 import numpy as np
 import torch.nn as nn
 import torch