tensorflow2.1 输出中间层结果
炼丹时经常需要可视化中间层的结果进行分析,本文采用tensorflow2.1版本,进行代码展示def show_middle(test_dataset_list, model, checkpoint_dir): tf.keras.backend.clear_session() #读取测试集 test_dataset = load_dataset_test() ...