diff --git a/config/default.yaml b/config/default.yaml index 0ee0b662..0c108b25 100644 --- a/config/default.yaml +++ b/config/default.yaml @@ -27,6 +27,7 @@ train: optimizer: Adam # Optimizer (Adam SGD) lr: 1.0e-3 # 学習率 (0 to 1.0 指数表記は末尾にe-数字) lr_gamma: 0.1 # SGD 用 ... Step Size 進んだ EPOCH で gammma が学習率に乗算される (0 to 1.0) + hate_mode: False # Enable hate mode specific configurations lr_momentum: 0.99 # SGD 用 ... モーメンタム... 今までの移動とこれから動くべき移動の平均をとり振動を防ぐための関数 (0 to 1.0) lr_step_size: 1000 # SGD 用 ... step size 学習率更新タイミングの EPOCH 数 (1 to num_epoch) num_epoch: 10000 # 試行回数 (lr_step_size to num_epoch) diff --git a/game_manager/game_manager.py b/game_manager/game_manager.py index d6c94638..1ad55da2 100644 --- a/game_manager/game_manager.py +++ b/game_manager/game_manager.py @@ -128,7 +128,7 @@ def __init__(self): self.game_time = args.game_time if args.mode in ("keyboard", "gamepad", "sample", "art", "train", "predict", "train_sample", "predict_sample", "train_sample2", "predict_sample2", "train_sample3", "predict_sample3"): self.mode = args.mode - if args.nextShapeMode in ("default", "hate"): + if args.nextShapeMode in ("normal", "hate"): self.nextShapeMode = args.nextShapeMode if args.drop_interval >= 0: self.drop_interval = args.drop_interval diff --git a/start.py b/start.py index 397cd608..00ec7875 100644 --- a/start.py +++ b/start.py @@ -18,7 +18,7 @@ def get_option(game_level, game_time, mode, nextShapeMode, random_seed, drop_int help='Specify mode (keyboard/gamepad/sample/art/train/predict/train_sample/predict_sample/train_sample2/predict_sample2/train_sample3/predict_sample3) if necessary') argparser.add_argument('--nextShapeMode', type=str, default=nextShapeMode, - help='Specify nextShapeMode (default/hate) if necessary') + help='Specify nextShapeMode (normal/hate) if necessary') argparser.add_argument('-r', '--random_seed', type=int, default=random_seed, help='Specify random seed if necessary')