toy_gnn.yaml 204 B

123456789
  1. # SPDX-License-Identifier: Apache-2.0
  2. _target_: models.gnn.ToyModel
  3. config:
  4. model_type: graph
  5. hidden_size: 128
  6. num_layers: 3
  7. defaults:
  8. - _self_
  9. - /trainer@_global_/trainer: ctltrainer