PyTorch
文章平均质量分 93
bujbujbiu
阿尔卑斯山的一棵草
展开
-
pytorch官方教程(详细版)
(一)构建神经网络BUILD THE NEURAL NETWORK使用pytorch构建神经网络进行FashionMNIST数据集中的图像分类原创 2022-04-10 10:19:52 · 31912 阅读 · 4 评论 -
pytorch实现CartPole-v1任务的DQN代码
DQN使用PyTorch在OpenAI Gym上的CartPole-v1任务上训练深度Q学习(DQN)智能体任务CartPole-v1环境中,手推车上面有一个杆,手推车沿着无摩擦的轨道移动。 通过对推车施加+1或-1的力来控制系统。 钟摆最开始为直立状态,训练的目的是防止其跌落。 杆保持直立的每个时间步长都提供+1的奖励。 当杆与垂直线的夹角超过15度时,或者推车从中心移出2.4个单位以上时,训练结束。智能体必须在两个动作之间做出决定-向左或向右移动推车-以便使与之相连的杆子保持直立。表现更好的方案将原创 2022-03-22 23:36:15 · 6173 阅读 · 5 评论 -
pytorch基础入门
pytorch基础入门,介绍了tensor,如何定义一个神经网络原创 2022-02-18 18:34:51 · 4570 阅读 · 2 评论 -
pytorch训练分类器
Training a Classifier前面学习到如何定义神经网络,计算损失并且对网络权重进行更新What about data?通常,当你必须处理图像,文本,音频或视频时,你可以使用能将数据加载到numpy数组的标准python包,然后将该数组转化成torch.*Tensor图像:Pillow, OpenCV音频:scipy,librosa文本:基于python或cython的原始加载,或者NLTK和SpaCy专门针对视觉,创建了名为torchvision的包,包含常见数据集(Imag原创 2022-02-22 21:16:54 · 2412 阅读 · 2 评论