diff --git a/PDMD/utils/_run.py b/PDMD/utils/_run.py
index eff0d08..dcf0501 100644
--- a/PDMD/utils/_run.py
+++ b/PDMD/utils/_run.py
@@ -5,6 +5,7 @@
 from torch.utils.data import ConcatDataset
 import os.path as osp
 import os
+import warnings
 import numpy as np
 import torch
 import random
@@ -12,11 +13,11 @@
 from ..models import ENERGY_Model, FORCE_Model
 from . import get_timestring, MutilWaterDataset, split_dataset, worker_init_fn, train, val, draw_two_dimension, reverse_min_max_scaler_1d
 
-
 # os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
 # os.environ['CUDA_VISIBLE_DEVICES'] = "0"
 # os.environ['CUDA_LAUNCH_BLOCKING'] = '0'
 # torch.set_default_dtype(torch.float64)
+warnings.filterwarnings('ignore', category=UserWarning, message='TypedStorage is deprecated')
 
 def run(config):
     np.random.seed(config.seed)