深度学习
肯定有问题
这个作者很懒,什么都没留下…
展开
-
upsample算子的调优(resnet18为例)
模型针对算子的性能调优原创 2022-08-11 10:15:20 · 585 阅读 · 1 评论 -
tensorflow网络
# 导入包import tensorflow as tffrom tensorflow_core.examples.tutorials.mnist import input_datafrom tensorflow.python.framework import graph_utilfrom npu_bridge.npu_init import *# 引入minist数据集mnist = input_data.read_data_sets('MNIST_data', one_hot=True原创 2022-02-23 16:43:44 · 725 阅读 · 0 评论 -
学习用torch写模型
用torch写一个topk+add的小模型import torchprint(torch.__version__)class Net1(torch.nn.Module): def __init__(self): super(Net1, self).__init__() def forward(self, x): output = torch.topk(x,1000) print("**************",type(outpu原创 2022-02-10 22:00:33 · 958 阅读 · 0 评论 -
tf的随笔
检测tf是否使用gpuimport numpy as npimport tensorflow as tfversion = tf.__version__gpu_ok = tf.test.is_gpu_available()print("tf version:",version,"\nuse GPU",gpu_ok)原创 2022-02-10 21:32:17 · 648 阅读 · 0 评论