提交 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): ...@@ -126,7 +126,8 @@ def build_dataloader(config, mode, device, use_dali=False, seed=None):
shuffle = config_sampler["shuffle"] shuffle = config_sampler["shuffle"]
else: else:
sampler_name = config_sampler.pop("name") 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: if "total_epochs" in sampler_argspec:
config_sampler.update({"total_epochs": epochs}) config_sampler.update({"total_epochs": epochs})
batch_sampler = eval(sampler_name)(dataset, **config_sampler) 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.
先完成此消息的编辑!
想要评论请 注册