我正在尝试运行一个名为Spacetimeformer的git repo,在训练一个程序时遇到了这个错误
lib\site-packages\torchmetrics\functional\text\bert.py", line 247, in <module>
AttributeError: module 'tqdm' has no attribute 'auto'
字符串
代码看起来像这样:
def _get_progress_bar(dataloader: DataLoader, verbose: bool = False) -> Union[DataLoader, tqdm.auto.tqdm]:
"""Helper function returning either the dataloader itself when `verbose = False`, or it wraps the dataloader with
`tqdm.auto.tqdm`, when `verbose = True` to display a progress bar during the embbeddings calculation."""
return tqdm.auto.tqdm(dataloader) if verbose else dataloader
型
我在网上得不到任何帮助,回购的作者也没有回答这个问题。我该如何解决这个问题?
1条答案
按热度按时间hi3rlvi21#
将
tqdm.auto.tqdm(dataloader)
替换为tqdm.tqdm(dataloader)
。