| 16 | |
| 17 | |
| 18 | def basic_static_model(): |
| 19 | ni = Input((None, 24, 24, 3)) |
| 20 | nn = Conv2d(16, (5, 5), (1, 1), padding='SAME', act=tf.nn.relu, name="conv1")(ni) |
| 21 | nn = MaxPool2d((3, 3), (2, 2), padding='SAME', name='pool1')(nn) |
| 22 | |
| 23 | nn = Conv2d(16, (5, 5), (1, 1), padding='SAME', act=tf.nn.relu, name="conv2")(nn) |
| 24 | nn = MaxPool2d((3, 3), (2, 2), padding='SAME', name='pool2')(nn) |
| 25 | |
| 26 | nn = Flatten(name='flatten')(nn) |
| 27 | nn = Dense(100, act=None, name="dense1")(nn) |
| 28 | nn = Dense(10, act=None, name="dense2")(nn) |
| 29 | M = Model(inputs=ni, outputs=nn, name='basic_static') |
| 30 | return M |
| 31 | |
| 32 | |
| 33 | class basic_dynamic_model(Model): |