- 博客(2)
- 收藏
- 关注
原创 猫狗大战挑战赛
建一个模型来完成 Kaggle 中的猫狗大战竞赛题目,有25000张标记好的猫和狗的图片用做训练,有12500张图片用做测试。 import numpy as np import matplotlib.pyplot as plt import os import torch import torch.nn as nn import torchvision from torchvision import models,transforms,datasets import time import json
2021-10-24 21:42:07
92
原创 pytorch基础练习和螺旋数据分类的代码练习
learning_rate = 1e-3 lambda_l2 = 1e-5 # nn 包用来创建线性模型 # 每一个线性模型都包含 weight 和 bias model = nn.Sequential( nn.Linear(D, H), nn.Linear(H, C) ) model.to(device) # 把模型放到GPU上 # nn 包含多种不同的损失函数,这里使用的是交叉熵(cross entropy loss)损失函数 criterion = torch.nn.CrossE.
2021-10-09 19:26:49
789
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人