keras
dataastron
工程师
展开
-
keras两种模型写法对比分析
1_绘图效果较为简洁 image_input = Input(shape=(224, 224, 3)) vision_model1 = ResNet50(include_top=False, weights='imagenet',input_tensor=Input(shape=(224, 224, 3))) encoded_image1 = vision_model1(image_input) x = GlobalAveragePooling2D(name='avg_pool')(encoded_imag原创 2020-05-09 00:55:46 · 606 阅读 · 0 评论 -
tf_keras内存控制
先查版本对症下药 import tensorflow as tf print(tf.__version__) tf1.x版本 from keras.backend.tensorflow_backend import set_session config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = ...原创 2020-04-27 20:20:25 · 5247 阅读 · 0 评论 -
keras实验1Keras框架搭建
实验介绍本文参考keras中文官方文档操作。传送门Keras框架搭建Keras中mnist数据集测试3行代码这么方便>>> git clone https://github.com/fchollet/keras.git >>> cd keras/examples/ >>> python mnist_mlp.py执行日志root@astron:/asdata# git clone https://gi原创 2017-09-16 03:08:05 · 919 阅读 · 0 评论 -
自定义lr代码
自定义学习率代码可以加快训练的速度,下面是代码的内容和学习率的截图。 import matplotlib.pyplot as plt def lrfn(epoch): LR_START = 0.00001 LR_MAX = 0.00005 *8#* strategy.num_replicas_in_sync LR_MIN = 0.00001 LR_RAMPUP_EP...原创 2020-03-04 22:10:57 · 558 阅读 · 0 评论 -
efficientnet.tfkeras
import efficientnet.tfkeras原创 2020-03-04 22:10:06 · 897 阅读 · 1 评论