TensorFlow
_小马奔腾
这个作者很懒,什么都没留下…
展开
-
TensorFlow 中遇到的坑
最近在做GANS的实验,用到了TensorFlow,把坑总结一下。1 ResourceExhaustedError 具体是这样的:现在主机上有四块卡,有一块有任务。 这也是TensorFlow很坑爹的一块:训练程序默认吃掉所有的显存。用掉所有的卡。。。解决办法 #CUDA_VISIBLE_DEVICES=1 python my_script.py 设置这个训练程序只可以看到一原创 2017-05-19 19:36:27 · 7035 阅读 · 0 评论 -
TensorFlow实战:Gans 学习明星人脸
EBGAN# -*- coding: utf-8 -*-"""Energy Based Generative Adversarial Networks (EBGAN)python version :2.7"""import osimport randomimport numpy as npimport tensorflow as tffrom PIL import Image#原创 2017-06-12 15:01:02 · 2032 阅读 · 0 评论