m0_58242646的博客

转载 ResNet18代码实现

import tensorflow as tf from tensorflow import keras from tensorflow.keras import layers, Sequential, Model, datasets, optimizers # 自定义的预处理函数 def preprocess(x, y): # 调用此函数时会自动传入x,y对象，shape为[b,28,28],[b] # 标准化到0-1 x = 2*tf.cast(x, dtype=t...