(args)
| 35 | |
| 36 | |
| 37 | def add_diffpool_args(args): |
| 38 | args.num_layers = 2 |
| 39 | args.num_pooling_layers = 1 |
| 40 | args.no_link_pred = False |
| 41 | args.pooling_ratio = 0.15 |
| 42 | args.embedding_dim = 64 |
| 43 | args.hidden_size = 64 |
| 44 | args.batch_size = 20 |
| 45 | args.dropout = 0.1 |
| 46 | return args |
| 47 | |
| 48 | |
| 49 | def add_gin_args(args): |
no outgoing calls
no test coverage detected