- 博客(1)
- 收藏
- 关注
原创 四折交叉验证训练函数
下面是具体的训练函数 其中要注意net.apply(weights_init)的位置要放在把模型移到gpu之后放在之前会导致训练函数直接退出。这是前期的准备工作因为在每一折前面我们需要清除模型的参数要用到net.apply(weights_init)所以要先定义这一块。
2024-03-25 22:33:18 212
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
下面是具体的训练函数 其中要注意net.apply(weights_init)的位置要放在把模型移到gpu之后放在之前会导致训练函数直接退出。这是前期的准备工作因为在每一折前面我们需要清除模型的参数要用到net.apply(weights_init)所以要先定义这一块。
2024-03-25 22:33:18 212
TA创建的收藏夹 TA关注的收藏夹
TA关注的人