Tensorflow
whyguu
这个作者很懒,什么都没留下…
展开
-
TensorFlow 模型框架 (tf.data.Dataset)
模型训练五步走(tf 1.7 直接运行):if __name__ == '__main__': # 1. data dataset = tf.data.Dataset.from_tensor_slices({'data': np.random.randn(6, 2), 'label'...原创 2018-04-07 00:25:02 · 621 阅读 · 0 评论 -
TensorFlow Multi-GPU demo
给出一个tensorflow多gpu的demo。用了minst 数据集,可直接运行。(tf1.8) 注:tensorflow gpu之间的操作好像有问题 import tensorflow as tf import numpy as np import os import argparse def arg_config(): parser = argparse.ArgumentPa...原创 2018-06-14 14:03:24 · 1339 阅读 · 0 评论 -
TensorFlow 实现 numpy 中的索引+赋值
# 将数组/Tensor中所有等于255的值修改成0 # 对于 numpy 数组 a a[a == 255] = 0 # 对于 tensorflow Tensor类 a a = tf.where(tf.not_equal(a, 255), a, tf.zeros_like(a))原创 2018-06-20 12:01:09 · 5742 阅读 · 0 评论