1 import tensorflow as tf
2 print(tf.__version__)
3
4
5 for gpu in tf.config.experimental.list_physical_devices('GPU'):
6 tf.config.experimental.set_memory_growth(gpu, True)
7
8
9 def nin_block(num_channels, kernel_size, strides, padding):
10 blk = tf.keras.models.Sequential()
11 blk.add(tf.keras.layers.Conv2D(num_channels, kernel_size, strides=strides,
12 padding=padding, activation='relu'))
13 blk.add(tf.keras.layers.Conv2D(num_channels, kernel_size=1, activation='relu'))
14 blk.add(tf.keras.layers.Conv2D(num_channels, kernel_size=1, activation='relu'))
15 return blk
16
17
18 net = tf.keras.models.Sequential()
19 net.add(nin_block(96, kernel_size=11, strides=4, padding='valid'))
20 net.add(tf.ke