Giter Site home page Giter Site logo

cifar10-cnn's Introduction

cifar10-CNN

卷积神经网络CNN在cifar10上的应用

本人进行了分别是在单个GPU和多个GPU下进行了测试,所以代码就被分成了两部分 单个GPU包含以下文件(在官方源码基础上进行更改),见博客卷积神经网络:CIFAR-10训练和测试(单块GPU):

读取本地CIFAR-10的二进制文件格式的内容
文件 作用
cifar10_model.py 建立卷积神经模型,定义损失函数,训练器和正确率计算等函数
cifar10_input.py
cifar10_train_eval.py 训练CIFAR-10和评估CIFAR-10模型

多个GPU包含以下文件(官方源码+本人注释和理解),见博客官方卷积神经网络cifar10源码的学习笔记(多块GPU):

文件 作用
input.py 读取本地CIFAR-10的二进制文件格式的内容。
cifar10.py 建立CIFAR-10的模型。
train.py 在CPU或GPU上训练CIFAR-10的模型。
eval.py 评估CIFAR-10模型的预测性能。
multi_gpu_train.py 在多GPU上训练CIFAR-10的模型。

cifar10-cnn's People

Contributors

skloismary avatar

Watchers

James Cloos avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.