修复Trainer里check_code函数忽略pin_memory参数导致的内存bug

Author: ouyhlan

fastNLP/core/tester.py

@@ -113,7 +113,7 @@ def __init__(self, data, model, metrics, batch_size=16, num_workers=0, device=No
         self.verbose = verbose
         self.use_tqdm = use_tqdm
         self.logger = logger
-        self.pin_memory = kwargs.get('pin_memory', True)
+        self.pin_memory = kwargs.get('pin_memory', False)
 
         if isinstance(data, DataSet):
             sampler = kwargs.get('sampler', None)

fastNLP/core/trainer.py

@@ -334,7 +334,6 @@ def on_epoch_end(self):
 except:
     from .utils import _pseudo_tqdm as tqdm
 import warnings
-from pkg_resources import parse_version
 
 from .batch import DataSetIter, BatchIter
 from .callback import CallbackManager, CallbackException, Callback
@@ -475,7 +474,7 @@ def __init__(self, train_data, model, optimizer=None, loss=None,
             if drop_last:
                 warnings.warn("drop_last is ignored when train_data is BatchIter.")
         # concerning issue from https://github.com/pytorch/pytorch/issues/57273
-        self.pin_memory = kwargs.get('pin_memory', False if parse_version(torch.__version__)==parse_version('1.9') else True)
+        self.pin_memory = kwargs.get('pin_memory', False)
         if isinstance(model, nn.parallel.DistributedDataParallel):  # 如果是分布式的
             # device为None
             if device is not None: