- 博客(4)
- 收藏
- 关注
原创 生成对抗网络(DCGAN, LSGAN, WGAN, WGAN-GP, SNGAN, RSGAN)TensorFlow实现
Paper:DCGAN: Unsupervised Representation Learning with Deep Convolutional Generative Adversarial NetworksWGAN: Wasserstein GANWGAN-GP: Improved Training of Wasserstein GANsLSGAN: Least Sq...
2018-07-25 11:46:26 6628 36
原创 TensorFlow只加载部分权重
var = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope="resnet_50/block1")saver = tf.train.Saver(var_list=var)saver.restore(sess, "resnet_50.ckpt")以上代码段为resnet_50的网络只加载name_scope为(resnet_5...
2018-07-25 10:33:21 3595
原创 GPU版TensorFlow怎么指定让CPU运行
由于某些原因GPU版的TensorFlow运行起来会出现一些问题,比如内存溢出等情况。此时我们可以用CPU和系统内存来运行我们的程序。代码如下:import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"os.environ["CUDA_VISIBLE_DEVICES"] = "-1"将以上三行代码放入程序首部即可。...
2018-07-25 10:05:41 9981 5
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人