![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
赵 XiaoQin
Never Say Never
展开
-
NMS各种变体
目标检测之非极大值抑制(NMS)各种变体转载 2020-11-30 09:54:25 · 348 阅读 · 0 评论 -
AlexNet模型代码整理
1. 网络模型 (论文是用两块gpu并行训练的,所以图中的模型是上下两部分的,第一层卷积核的个数是48+48=96) 2. 模型参数 3. 模型代码 import torch.nn as nn import torch class AlexNet(nn.Module): def __init__(self, num_classes=1000, init_weights=False): super(AlexNet, self).__init__() .原创 2020-07-05 20:28:32 · 1144 阅读 · 0 评论 -
pytorch笔记8--optimizer
对比各种优化器的效果 数据分布如下图: import torch import torch.utils.data as Data import torch.nn.functional as Func from matplotlib import pyplot as plt torch.manual_seed(1) #hyper parameters LR=0.01 BATCH_SIZ...原创 2020-03-24 23:36:07 · 166 阅读 · 0 评论 -
深度学习笔记--Kaggle比赛之房价预测
#获取和读取数据 import torch import torch.nn as nn import pandas as pd #处理数据 import All_function as func #自定义包 torch.set_default_tensor_type(torch.FloatTensor) train_data=pd.read_csv('Kaggle_...原创 2019-11-26 15:34:50 · 821 阅读 · 1 评论 -
深度学习笔记--Fashion_mnist+softmax的简单实现
1. 主文件 import torch import All_function as func import torch.nn as nn #获取数据 batch_size=256 train_iter,test_iter=func.load_data_fashion_minist(256) #定义和初始化模型 class LinearNet(nn.Module): def __i...原创 2019-11-13 11:59:35 · 468 阅读 · 0 评论