Four samples by default
@@ -49,7 +49,7 @@ if __name__ == "__main__":
parser.add_argument("--cuda", action='store_true', default=False, help="Enable cuda")
parser.add_argument("-n", "--name", required=True, help="Name of the run")
parser.add_argument("-l", "--load", required=True, help="Load model and continue in RL mode")
- parser.add_argument("--samples", type=int, default=1, help="Count of samples in prob mode")
+ parser.add_argument("--samples", type=int, default=4, help="Count of samples in prob mode")
args = parser.parse_args()
saves_path = os.path.join(SAVES_DIR, args.name)
