garage.tf.models.categorical_cnn_model moduleΒΆ