在overstack flow上看到有人说是因为版本问题，（这个命令在torch.1.9以上），但下面有人反映说这个不行；

解决方案：

将代码替换，将torch.inference_mode()替换为torch.no_grad()

@torch.no_grad()
# @torch.inference_mode()
    with torch.no_grad():
    # with torch.inference_mode():

# 深度学习 # 神经网络 # 算法
Logo
2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

cover

ImagineNav++：通过场景想象激发视觉语言模型，使其成为具身导航器

avatar 2048 AI社区
cover

Flutter for OpenHarmony 实战：疯狂头像 App（四）— 通义万相 AIGC 联调与相册持久化实战

avatar 2048 AI社区

【Mysql基础（二）】函数、约束、多表查询与事务

外键约束（Foreign Key Constraint）是关系型数据库中用于维护表间引用完整性的机制。它确保一个表（子表）中的列值必须匹配另一个表（父表）中的主键或唯一键值，从而建立表间的关联关系。事务是数据库操作的最小逻辑单元，保证一组操作要么全部成功，要么全部失败。MySQL 支持事务的存储引擎包括 InnoDB 和 NDB，而 MyISAM 不支持事务。），每条 SQL 语句作为一个独立事务

avatar 2048 AI社区