import tensorflow as tf


def cnn_model(x, amp_factor=1):
    with tf.variable_scope('model'):
        conv1 = tf.layers.conv2d(x, filters=32*amp_factor, kernel_size=[5, 3],
                                 data_format='channels_last', padding= "same",
                                 strides=(2, 1),
