torch0.4 volatile was removed and now has no effect. Use `with torch.no_grad():` instead.

volatile was removed and now has no effect. Use with torch.no_grad(): instead.

  molded_images = Variable(molded_images, volatile=True)

修改为

    with torch.no_grad():
        molded_images = Variable(molded_images)

猜你喜欢

转载自blog.csdn.net/qq_35608277/article/details/85339538
今日推荐