问题描述:debug时查看变量(variables)卡顿,读取缓慢,不显示 。(Variables只显示Connected,卡住不动) 这种情况下你会发现单步调试也卡住了,看不到执行之后的情况。

可能原因:代码中使用了多线程,例如multiprocessing或者pytorch读取数据时将num_workers设置的值大于0(如下所示)

    train_loader = torch.utils.data.DataLoader(train_split,
        batch_size=batch_size, shuffle=True, num_workers=2)
    val_loader = torch.utils.data.DataLoader(val_split,
        batch_size=batch_size, shuffle=False, num_workers=2)

解决方法: 多线程改为单线程,例如将pytorch中num_workers设置为0

    train_loader = torch.utils.data.DataLoader(train_split,
        batch_size=batch_size, shuffle=True, num_workers=0)
    val_loader = torch.utils.data.DataLoader(val_split,
        batch_size=batch_size, shuffle=False, num_workers=0)
Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐