- 博客(2)
- 收藏
- 关注
原创 PyTorch中optim的用法
在中,我们可以使用各种类型的优化器来更新神经网络模型的参数。这些优化器通常被定义为类,并通过指定要优化的参数和学习率等超参数来进行初始化。
2023-05-22 00:14:47
282
1
原创 torch.nn的用法
torch.nn提供了各种类型的神经网络层,如全连接层、卷积层、循环神经网络等。这些层可以通过继承类来自定义,也可以直接使用torch.nn模块中提供的预定义层。
2023-05-22 00:02:52
802
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人