import tensorflow as tf

    self.output_dir = paths.trial(paths.experiment(constants.EXPERIMENT_PATH, 'big_two_layer'), trial)

  def train_once(self, iteration, presets=None, masks=None):
    tf.reset_default_graph()
    sess = tf.Session()
    dataset = dataset_mnist.DatasetMnist(
