# ----------------------------------------------------------------------------- # Set information about the data set # ----------------------------------------------------------------------------- th.max_level = 10 th.volume_only = True th.developer_code = "use_log" th.input_shape = [20] th.num_classes = 3 th.output_dim = 3 th.target_dim = 1 th.target_dtype = tf.int32 th.val_size = 1 th.loss_string = "cross_entropy" # ----------------------------------------------------------------------------- # Set common trainer configs # ----------------------------------------------------------------------------- th.epoch = 1000 # STOCK 1000 th.gather_note = True th.export_tensors_upon_validation = True th.train = True th.overwrite = True th.save_model = True th.save_mode = SaveMode.ON_RECORD th.validation_per_round = 2 th.early_stop = True
# ----------------------------------------------------------------------------- # Set information about the data set # ----------------------------------------------------------------------------- th.max_level = 10 th.volume_only = True th.developer_code = 'use_log' th.input_shape = [20] th.num_classes = 3 th.output_dim = 3 th.target_dim = 1 th.target_dtype = tf.int32 th.val_size = 1 th.loss_string = 'cross_entropy' # ----------------------------------------------------------------------------- # Set common trainer configs # ----------------------------------------------------------------------------- th.epoch = 1000 th.gather_note = True th.export_tensors_upon_validation = True th.train = True th.overwrite = True th.save_model = True th.save_mode = SaveMode.ON_RECORD th.validation_per_round = 2 th.early_stop = True