2023年4月21日
pytorch训练模型时的小问题
摘要： 数据导入时，如果遇到数据少于要读取的batch size，可能会出现错误，此时可以在DataLoader中使用drop_last = True进行丢弃 阅读全文
posted @ 2023-04-21 13:20 燕汀 阅读(16) 评论(0) 推荐(0)