Normalization and transforms for batch_to_data class

2020-06-15 15:14:08 +02:00
parent bc70f42c74
commit 4898e98851
8 changed files with 26 additions and 24 deletions
--- a/_parameters.py
+++ b/_parameters.py
@ -21,6 +21,7 @@ main_arg_parser.add_argument("--project_neptune_key", type=str, default=os.geten

 # Data Parameters
 main_arg_parser.add_argument("--data_worker", type=int, default=10, help="")
+main_arg_parser.add_argument("--data_sampling_k", type=int, default=1024, help="")
 main_arg_parser.add_argument("--data_root", type=str, default='data', help="")
 main_arg_parser.add_argument("--data_dataset_type", type=str, default='GridClusters', help="")
 main_arg_parser.add_argument("--data_use_preprocessed", type=strtobool, default=True, help="")