pytorch 属性错误:模块“tqdm”没有属性“auto”

jgwigjjp  于 2023-08-05  发布在  其他
关注(0)|答案(1)|浏览(456)

我正在尝试运行一个名为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


我在网上得不到任何帮助,回购的作者也没有回答这个问题。我该如何解决这个问题?

hi3rlvi2

hi3rlvi21#

tqdm.auto.tqdm(dataloader)替换为tqdm.tqdm(dataloader)

相关问题