Commit 7dc4fed3 authored by Jiri Borovec's avatar Jiri Borovec

fix args

parent 6f452843
......@@ -53,7 +53,7 @@ DEFAULT_PARAMS = {
'max_iter': 100,
'gc_regul': 1e-2,
'nb_labels': 15,
'nb_runs': 25, # 500
'nb_runs': NB_THREADS, # 500
'gc_reinit': True,
'ptn_split': True,
'overlap_mj': False,
......@@ -121,7 +121,7 @@ def create_args_parser(dict_params):
parser.add_argument('--nb_jobs', type=int, required=False, default=NB_THREADS,
help='number of processes running in parallel')
parser.add_argument('--methods', type=int, required=False, nargs='+', default=None,
parser.add_argument('--methods', type=str, required=False, nargs='+', default=None,
help='possible APD methods', choices=['PCA', 'ICA', 'DL'])
return parser
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment