# Device configurations # ----------------------------------------------------------------------------- th.allow_growth = False th.gpu_memory_fraction = 0.30 # ----------------------------------------------------------------------------- # Set information about the data set # ----------------------------------------------------------------------------- th.input_shape = [28, 28, 1] th.num_classes = 10 # ----------------------------------------------------------------------------- # Set common trainer configs # ----------------------------------------------------------------------------- th.early_stop = True th.patience = 6 th.shuffle = True th.save_model = False th.save_mode = SaveMode.ON_RECORD th.overwrite = True th.gather_note = True th.print_cycle = 5 th.validation_per_round = 2 th.val_batch_size = 1000 th.evaluate_train_set = True th.evaluate_val_set = True th.evaluate_test_set = True
# ----------------------------------------------------------------------------- # 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 th.patience = 10 th.lives = 1 th.clip_threshold = 1.0 th.clip_method = "value" th.print_cycle = 50 th.val_progress_bar = True th.val_batch_size = -1 th.val_num_steps = 2000 th.clip_nan_protection = False th.state_nan_protection = False th.terminate_on_nan = True