diff --git a/examples/neural_architecture_search/neural_architecture_search.py b/examples/neural_architecture_search/neural_architecture_search.py
index 90705a3..3e49376 100644
--- a/examples/neural_architecture_search/neural_architecture_search.py
+++ b/examples/neural_architecture_search/neural_architecture_search.py
@@ -219,9 +219,8 @@ def __len__(self):
         )
 
         logging.info("train_acc %f , train_obj %f ", train_acc, train_obj)
-        utils.save(model, os.path.join(args.save, "weights.pt"))
         print("epoch_step_time:", epoch_time)
-        utils.save(model, os.path.join(args.save, "weights.pt"))
+        utils.save(model, os.path.join(args.save, "weights.pkl"))
     print("average_step_time", average_runtime / 3)
 
 def train(train_queue, valid_queue, model, criterion, optimizer, boat_optimizer):
diff --git a/examples/neural_architecture_search/utils.py b/examples/neural_architecture_search/utils.py
index 117a0b1..f15ffc2 100644
--- a/examples/neural_architecture_search/utils.py
+++ b/examples/neural_architecture_search/utils.py
@@ -155,10 +155,10 @@ def count_parameters_in_MB(model):
     return np.sum(np.prod(v.shape) for name, v in model.named_parameters() if "auxiliary" not in name) / 1e6
 
 def save_checkpoint(state, is_best, save):
-    filename = os.path.join(save, "checkpoint.pth.tar")
+    filename = os.path.join(save, "checkpoint.pkl")
     jt.save(state, filename)
     if is_best:
-        best_filename = os.path.join(save, "model_best.pth.tar")
+        best_filename = os.path.join(save, "model_best.pkl")
         shutil.copyfile(filename, best_filename)
 
 def save(model, model_path):
@@ -184,4 +184,4 @@ def create_exp_dir(path, scripts_to_save=None):
         os.mkdir(os.path.join(path, "scripts"))
         for script in scripts_to_save:
             dst_file = os.path.join(path, "scripts", os.path.basename(script))
-            shutil.copyfile(script, dst_file)
\ No newline at end of file
+            shutil.copyfile(script, dst_file)