qq_27009517的博客

最近在使用perceptual_loss过程中，出现了模型随着训练占用的内存越来越大的问题，千万不要在遍历数据的过程中使用tf的相关操作，必要的操作放到模型中，如果在模型外执行，会不断增加新的节点，导致网络占用的内存随着训练的进行越来越大tf2预训练好的模型加载放在__init__中，返回函数名。reference：https://blog.youkuaiyun.com/qq_37876289/article/details/106157359...

/home/hlx/anaconda3/bin/python3.6 /home/hlx/OCR/ctpn_1109/abcd.py2.2.02020-11-11 09:52:49.803557: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library libcuda.so.12020-11-11 09:52:49.835435: E tensorflow...

下载地址：https://download.microsoft.com/download/3/E/1/3E1C3F21-ECDB-4869-8368-6DEBA77B919F/kagglecatsanddogs_3367a.zip

# coding: utf-8# @Time :20-11-9 上午9:47# @Author :LUAN XIN HUANG# @Modifer :# @FileName :trainsfer_learning.pyimport numpy as npimport tensorflow as tffrom tensorflow import keras# 全连接网络的权重# layer = keras.layers.Dense(3)# layer.bui

导入包import numpy as npimport tensorflow as tffrom tensorflow import kerasfrom tensorflow.keras import layersKeras中有三个内置的RNN层：keras.layers.SimpleRNN：全连接的rnn，前一个timestep的输出传递到下一个timestep。keras.layers.GRU：在Cho et al., 2014提出.keras.layers.LSTM：在Hochre

reference：https://tensorflow.google.cn/guide/keras/transfer_learning