摘要: 作者:Henry链接:https://zhuanlan.zhihu.com/p/539233251来源:知乎 1、从模型方面考虑。举例说明:本身问题是二次的,用线性模型处理问题就是欠拟合,用三次及更高次处理问题就是过拟合。但是这里未考虑数据量的多少,只是针对本身模型阶次的考虑。而且现实问题,越强大的 阅读全文
posted @ 2022-12-05 20:16 Tomorrow1126 阅读(333) 评论(0) 推荐(0) 编辑
摘要: torch.cuda.is_available()cuda是否可用; torch.cuda.device_count()返回gpu数量; torch.cuda.get_device_name(0)返回gpu名字,设备索引默认从0开始; torch.cuda.current_device()返回当前设 阅读全文
posted @ 2022-12-05 20:15 Tomorrow1126 阅读(303) 评论(0) 推荐(0) 编辑
摘要: 1.lr_scheduler.StepLR class torch.optim.lr_scheduler.StepLR(optimizer, step_size, gamma=0.1, last_epoch=-1) 功能: 等间隔调整学习率,调整倍数为gamma倍,调整间隔为step_size。间隔 阅读全文
posted @ 2022-12-05 11:23 Tomorrow1126 阅读(199) 评论(0) 推荐(0) 编辑