pytorch训练模型时的小问题

数据导入时，如果遇到数据少于要读取的batch size，可能会出现错误，此时可以在DataLoader中使用drop_last = True进行丢弃