- 博客(22)
- 收藏
- 关注
原创 val_check_interval
设置为 100,那么每经过 100 个训练步骤,Trainer 就会在验证集上评估一次模型,然后输出验证集上的性能指标。这个过程不会影响训练,仅仅是用来监控模型在验证集上的表现。设置为 1,也就是每个训练步骤都会在验证集上评估一次模型。这样会非常耗时,因此在实际训练中,我们通常会将。调大一些,比如设置为 10 或 100,以平衡验证集上的性能监控和训练速度。类的一个参数,它用于控制训练过程中在验证集上评估模型的频率。指定了多少个训练步骤之后,Trainer 会调用模型的。方法来计算在验证集上的性能指标。
2023-04-13 16:13:01
883
原创 gradient_clip_val
是PyTorch Lightning中的一个训练器参数,用于控制梯度的裁剪(clipping)。梯度裁剪是一种优化技术,用于防止梯度爆炸(gradient explosion)和梯度消失(gradient vanishing)问题,这些问题会影响神经网络的训练过程。,则所有的梯度将会被裁剪到1.0范围内,这可以避免梯度爆炸的问题。参数的值表示要将梯度裁剪到的最大范数值。如果梯度的范数超过这个值,就会对梯度进行裁剪,将其缩小到指定的范围内。这将创建一个带有梯度裁剪功能的训练器,最大裁剪范围为0.5。
2023-04-13 16:11:03
1284
2
原创 torch_geometric.utils.remove_self_loops()
torch_geometric.utils.remove_self_loops()
2023-04-11 17:16:05
497
原创 torch.sparse_coo_tensor函数根据索引对生成邻接矩阵
torch.sparse_coo_tensor函数根据索引对生成邻接矩阵
2023-04-06 21:53:35
572
原创 torch_geometric的whl文件安装链接
注:先pip安装torch_sparse等文件,最后pip install torch_geometric。
2023-04-06 15:48:19
97
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人