提交 37795086 编写于 作者: G gaotingquan 提交者: cuicheng01

getargspec -> getfullargspec

getargspec dont support param annotations
上级 7e0207e5
......@@ -126,7 +126,8 @@ def build_dataloader(config, mode, device, use_dali=False, seed=None):
shuffle = config_sampler["shuffle"]
else:
sampler_name = config_sampler.pop("name")
sampler_argspec = inspect.getargspec(eval(sampler_name).__init__).args
sampler_argspec = inspect.getfullargspec(eval(sampler_name)
.__init__).args
if "total_epochs" in sampler_argspec:
config_sampler.update({"total_epochs": epochs})
batch_sampler = eval(sampler_name)(dataset, **config_sampler)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册