学习
刘二大人《PyTorch深度学习实践》
B站视频:B站视频
问题:
1.数据准备阶段
要注意数据的维度
2.设计模型
Design model using Class!
3.构造Loss函数和优化函数
4.重复训练
import torch
import torch.nn as nn
x_data = torch.Tensor([[1.0], [2.0], [3.0]])
y_data = torch.Tensor([[2.0], [
刘二大人《PyTorch深度学习实践》
B站视频:B站视频
问题:
1.数据准备阶段
要注意数据的维度
2.设计模型
Design model using Class!
3.构造Loss函数和优化函数
4.重复训练
import torch
import torch.nn as nn
x_data = torch.Tensor([[1.0], [2.0], [3.0]])
y_data = torch.Tensor([[2.0], [