快盘下载:好资源、好软件、快快下载吧!

快盘排行|快盘最新

当前位置:首页软件教程电脑软件教程 → PyTorch-Error:-Expected-more-than-1-value-per-channel-when-training

PyTorch-Error:-Expected-more-than-1-value-per-channel-when-training

时间:2022-09-07 10:20:21人气:作者:快盘下载我要评论

很简单

使用 DataLoader 增加一个参数 drop_last=True

self.train_loader = DataLoader(self.train_set,
                               batch_size=self.batch_size,
                               shuffle=True, 
                               pin_memory=False,
                               num_workers=self.num_workers,
                               collate_fn=collater,
                               drop_last=True
                              )

相关文章

网友评论

快盘下载暂未开通留言功能。

关于我们| 广告联络| 联系我们| 网站帮助| 免责声明| 软件发布

Copyright 2019-2029 【快快下载吧】 版权所有 快快下载吧 | 豫ICP备10006759号公安备案:41010502004165

声明: 快快下载吧上的所有软件和资料来源于互联网,仅供学习和研究使用,请测试后自行销毁,如有侵犯你版权的,请来信指出,本站将立即改正。