diff --git a/PyTorch/contrib/cv/classification/SPACH/main.py b/PyTorch/contrib/cv/classification/SPACH/main.py
index e641f0749956099930a1f363c34a57ae8ff747bc..93fd95f00b65fe895394a17f4e4da92f4659f4b4 100644
--- a/PyTorch/contrib/cv/classification/SPACH/main.py
+++ b/PyTorch/contrib/cv/classification/SPACH/main.py
@@ -1,10 +1,22 @@
-# Copyright (c) 2015-present, Facebook, Inc.
-# All rights reserved.
+# Copyright (c) 2019-2021 NVIDIA CORPORATION. All rights reserved.
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 import argparse
 import datetime
 import numpy as np
 import time
 import torch
+import torch_npu
 import torch.backends.cudnn as cudnn
 import json
 import os