编码和解码图片pytorch

作者: 一瓶泡沫
发布时间: 2024-12-13 05:11:20 (2月前)
转自：

2 条回复

0#
回复此人
听风～ | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 1）更换线路 </p> <pre> <code> mse_loss = loss(torch.tensor(decoded_x, dtype=torch.float), x) </code> </pre> <P> 用线 </p> <pre> <code> mse_loss = loss(decoded_x, x) </code> </pre> <P> 2）更换线 </p> <pre> <code> model.eval() with torch.no_grad(): for x, y in train_loader: </code> </pre> <P> 用线条 </p> <P> 替换线 </p> <pre> <code> model.eval() with torch.no_grad(): for x, y in test_loader: </code> </pre> </DIV>

编辑

登录后才能参与评论