steffen/point_to_primitive
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Grid Clusters.

Browse Source
This commit is contained in:
Si11ium
2020-06-07 16:47:52 +02:00
parent 8d0577b756
commit 2a767bead2
14 changed files with 278 additions and 149 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
_parameters.py
View File

@ -40,7 +40,8 @@ main_arg_parser.add_argument("--train_sto_weight_avg", type=strtobool, default=F
main_arg_parser.add_argument("--train_opt_reset_interval", type=strtobool, default=False, help="")
# Model
main_arg_parser.add_argument("--model_type", type=str, default="PN2", help="")
# Possible Model arguments are: P2P, PN2, P2G
main_arg_parser.add_argument("--model_type", type=str, default="P2G", help="")
main_arg_parser.add_argument("--model_norm_as_feature", type=strtobool, default=True, help="")
main_arg_parser.add_argument("--model_activation", type=str, default="leaky_relu", help="")