人工智能
july_young
外包8年,想进大厂。
展开
-
conda 安装nni automl
#!/bin/bashnni=$(conda env list |grep nni|awk '{print $1}')if [ ! -z $nni ];then exit 0ficonda create -n nni python=3.6 -ysource activate nniconda install cudatoolkit==10.1 -yconda install cudnn==7.6 -ypip install tensorflow==1.9 -y...原创 2020-12-10 11:13:35 · 859 阅读 · 0 评论 -
pytorch 手写数字图片识别代码
1、训练文件import torchfrom torch import nnfrom torch.nn import functional as Ffrom torch import optimimport torchvisionimport matplotlib.pyplot as plt# 辅助函数def plot_curve(data): fig = plt.figure() plt.plot(range(len(data)), data, color='b原创 2020-12-02 14:43:42 · 1013 阅读 · 3 评论 -
pytorch 线性回归代码
import torchfrom matplotlib import pyplot as pltimport numpy as np# 样本数:1000num_examples = 1000# 设置权重true_w = 3# 设置偏置true_b = 20# 1000个特征数为2的样本delta_x = torch.randn(num_examples, dtype=torch.float32)meta_x = torch.arange(0, num_examples, dty.原创 2020-12-01 17:27:27 · 371 阅读 · 0 评论