test/fastNLP
1
0
Fork 0
mirror of https://gitee.com/fastnlp/fastNLP.git synced 2024-11-29 18:59:01 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修正 _TruncatedDataLoader 2

Browse Source
This commit is contained in:
YWMditto 2022-08-08 20:32:52 +08:00
parent e86645380a
commit 0033219fe1
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
fastNLP/core/controllers/utils/utils.py
View File

@ -132,9 +132,16 @@ class _TruncatedDataLoader:
def __getattr__(self, item):
return getattr(self.dataloader, item)
# def __setattr__(self, key, value):
# # 添加该函数使得在进行实验性训练或者评测时，用户对于 trainer.dataloader 的感觉和正常训练完全一样；
# setattr(self.dataloader, key, value)
def __setattr__(self, key, value):
# 添加该函数使得在进行实验性训练或者评测时，用户对于 trainer.dataloader 的感觉和正常训练完全一样；
# 注意这里不能直接 ``setattr(self.dataloader, key, value)``，会导致死循环；
if "dataloader" in self.__dict__:
if hasattr(self.dataloader, key):
setattr(self.dataloader, key, value)
else:
self.__dict__[key] = value
else:
self.__dict__[key] = value
def check_evaluate_every(evaluate_every):