(args)
| 47 | |
| 48 | |
| 49 | def add_gin_args(args): |
| 50 | args.epsilon = 0.0 |
| 51 | args.hidden_size = 32 |
| 52 | args.num_layers = 5 |
| 53 | args.num_mlp_layers = 2 |
| 54 | args.train_epsilon = True |
| 55 | args.pooling = "sum" |
| 56 | args.batch_size = 128 |
| 57 | return args |
| 58 | |
| 59 | |
| 60 | def add_sortpool_args(args): |
no outgoing calls
no test coverage detected