All Projects → wamuir → golang-tf

wamuir / golang-tf

Licence: MIT license
Working golang + tensorflow

Programming Languages

Dockerfile
14818 projects
python
139335 projects - #7 most used programming language
shell
77523 projects
HCL
1544 projects

Projects that are alternatives of or similar to golang-tf

tensorflow 2.0 tutorial
tensorflow 2.0 正式版实用教程/tutorial
Stars: ✭ 48 (+128.57%)
Mutual labels:  tensorflow2
checkmate
Training neural networks in TensorFlow 2.0 with 5x less memory
Stars: ✭ 116 (+452.38%)
Mutual labels:  tensorflow2
3D-GuidedGradCAM-for-Medical-Imaging
This Repo containes the implemnetation of generating Guided-GradCAM for 3D medical Imaging using Nifti file in tensorflow 2.0. Different input files can be used in that case need to edit the input to the Guided-gradCAM model.
Stars: ✭ 60 (+185.71%)
Mutual labels:  tensorflow2
TF2DeepFloorplan
TF2 Deep FloorPlan Recognition using a Multi-task Network with Room-boundary-Guided Attention. Enable tensorboard, quantization, flask, tflite, docker, github actions and google colab.
Stars: ✭ 98 (+366.67%)
Mutual labels:  tensorflow2
pyradox
State of the Art Neural Networks for Deep Learning
Stars: ✭ 61 (+190.48%)
Mutual labels:  tensorflow2
E2E-Object-Detection-in-TFLite
This repository shows how to train a custom detection model with the TFOD API, optimize it with TFLite, and perform inference with the optimized model.
Stars: ✭ 28 (+33.33%)
Mutual labels:  tensorflow2
Imgclsmob
Sandbox for training deep learning networks
Stars: ✭ 2,405 (+11352.38%)
Mutual labels:  tensorflow2
Bearcat captcha
熊猫识别不定长验证码,基于tensorflow2.2(tensorflow2.3也可以运行)轻松就能练出不错的模型
Stars: ✭ 67 (+219.05%)
Mutual labels:  tensorflow2
transformer
Build English-Vietnamese machine translation with ProtonX Transformer. :D
Stars: ✭ 41 (+95.24%)
Mutual labels:  tensorflow2
TensorFlow2.0 SSD
A tensorflow_2.0 implementation of SSD (Single Shot MultiBox Detector) .
Stars: ✭ 83 (+295.24%)
Mutual labels:  tensorflow2
Grokking-Machine-Learning
This repo aims to contain different machine learning use cases along with the descriptions to the model architectures
Stars: ✭ 54 (+157.14%)
Mutual labels:  tensorflow2
AiSpace
AiSpace: Better practices for deep learning model development and deployment For Tensorflow 2.0
Stars: ✭ 28 (+33.33%)
Mutual labels:  tensorflow2
word-embeddings-from-scratch
Creating word embeddings from scratch and visualize them on TensorBoard. Using trained embeddings in Keras.
Stars: ✭ 22 (+4.76%)
Mutual labels:  tensorflow2
DataAugmentationTF
Implementation of modern data augmentation techniques in TensorFlow 2.x to be used in your training pipeline.
Stars: ✭ 35 (+66.67%)
Mutual labels:  tensorflow2
Open-Source-Models
Address book for computer vision models.
Stars: ✭ 30 (+42.86%)
Mutual labels:  tensorflow2
Chatbot
一个可以自己进行训练的中文聊天机器人, 根据自己的语料训练出自己想要的聊天机器人,可以用于智能客服、在线问答、智能聊天等场景。目前包含seq2seq、seqGAN版本、tf2.0版本、pytorch版本。
Stars: ✭ 2,441 (+11523.81%)
Mutual labels:  tensorflow2
tensorflow-maml
TensorFlow 2.0 implementation of MAML.
Stars: ✭ 79 (+276.19%)
Mutual labels:  tensorflow2
potato-disease-classification
Potato Disease Classification - Training, Rest APIs, and Frontend to test.
Stars: ✭ 95 (+352.38%)
Mutual labels:  tensorflow2
gans-2.0
Generative Adversarial Networks in TensorFlow 2.0
Stars: ✭ 76 (+261.9%)
Mutual labels:  tensorflow2
manning tf2 in action
The official code repository for "TensorFlow in Action" by Manning.
Stars: ✭ 61 (+190.48%)
Mutual labels:  tensorflow2

golang-tf

About

This repository contains Dockerfiles for using Tensorflow in Go. You can build on your own or pull an image from wamuir/golang-tf on Docker Hub:

docker pull wamuir/golang-tf

Example Usage

See the example Dockerfile.

GPU Support

Installation of the NVIDIA Container Toolkit is necessary for GPU use. Then, expose the GPU(s) to the docker container using the --gpus all flag. Example:

docker run -it --rm --gpus all wamuir/golang-tf:gpu

Background

For background on the problem, see Tensorflow issues 39307, 39744, 41808, 43847, and 44655, among others.

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].