def METHOD_NAME():
    image_shape = (1, 4, 84, 84)
    mod, params = testing.dqn.get_workload(batch_size=1, image_shape=image_shape)
    benchmark_execution(mod, params, data_shape=image_shape, out_shape=(1, 18))