摘要:
import torch import torch.nn as nn import torch.optim as optim import random 1. 生成训练数据 def generate_data(max_n=5, num_samples=1000): data = [] for _ i 阅读全文
摘要:
import os import torch import torch.nn as nn import torch.nn.functional as F from torchsummary import summary from torchvision import transforms from 阅读全文
摘要:
import torch import torch.nn as nn import torch.optim as optim import torchvision import torchvision.transforms as transforms import matplotlib.pyplot 阅读全文