Browse Source

为 _TruncatedDataLoader 添加 __setattr__

tags/1.0.0beta2
YWMditto 2 years ago
parent
commit
d7ce92975c
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      fastNLP/core/controllers/utils/utils.py

+ 4
- 0
fastNLP/core/controllers/utils/utils.py View File

@@ -132,6 +132,10 @@ class _TruncatedDataLoader:
def __getattr__(self, item): def __getattr__(self, item):
return getattr(self.dataloader, item) return getattr(self.dataloader, item)


def __setattr__(self, key, value):
# 添加该函数使得在进行实验性训练或者评测时,用户对于 trainer.dataloader 的感觉和正常训练完全一样;
setattr(self.dataloader, key, value)



def check_evaluate_every(evaluate_every): def check_evaluate_every(evaluate_every):
r""" r"""


Loading…
Cancel
Save