IOS、Python自学心得30天 Day-9 模组训练改善-1

前言:
之前都是一次性训练好模型，但是一旦资料太多要训练，动辄就是数小时，甚至几天
所以我打算改善训练的过程，让它能储存上次进度来训练模组

程序码:

import os
import tensorflow as tf
from tensorflow import keras
from tensorflow.python.keras import backend as K
from tensorflow.python.keras.models import Model
from tensorflow.python.keras.layers import Flatten, Dense, Dropout
from tensorflow.python.keras.applications.resnet import ResNet50
from tensorflow.python.keras.optimizer_v2.adam import Adam
from tensorflow.python.keras.preprocessing.image import ImageDataGenerator
from tensorflow.python.keras.models import Sequential, load_model
from tensorflow.python.keras.layers import LSTM, Dropout, Dense
from tensorflow.python.keras.callbacks import ModelCheckpoint
from tensorflow.python.framework.ops import disable_eager_execution

disable_eager_execution()

tf.compat.v1.experimental.output_all_intermediates(True)

checkpoint_path = "training_1/cp.ckpt"
checkpoint_dir = os.path.dirname(checkpoint_path)

# Create a callback that saves the model's weights
cp_callback = tf.keras.callbacks.ModelCheckpoint(filepath=checkpoint_path,
                                                 save_weights_only=True,
                                                 verbose=1)

# 输出整个网路结构
print(net_final.summary())

# 训练模型
net_final.fit(train_batches,
              steps_per_epoch = train_batches.samples // BATCH_SIZE,
              validation_data = valid_batches,
              validation_steps = valid_batches.samples // BATCH_SIZE,
              epochs = NUM_EPOCHS,
              callbacks = [cp_callback])

os.listdir(checkpoint_dir)