wandb_version: 1

_wandb:
  desc: null
  value:
    cli_version: 0.13.9
    framework: torch
    is_jupyter_run: false
    is_kaggle_kernel: false
    python_version: 3.10.9
    start_time: 1690916267.460289
    t:
      1:
      - 1
      - 5
      - 53
      - 55
      - 77
      2:
      - 1
      - 5
      - 53
      - 55
      - 77
      3:
      - 1
      - 16
      - 23
      4: 3.10.9
      5: 0.13.9
      8:
      - 5
input_path:
  desc: null
  value: null
location_data:
  desc: null
  value:
    input_path:
    transfer_path: null
model:
  desc: null
  value: null
model_config:
  desc: null
  value:
    activation: relu
    conv_type: FastRGCNConv
    device: cuda
    dropout: 0.01
    edge_encoder_dim: 2
    edge_ext_thresh: 0.25
    hidden_dim: 64
    layer_spec: null
    load_model: false
    log_graphs: true
    lstm_input_dim: 50
    lstm_output_dim: 20
    model: ss_rs2g
    model_load_path: ''
    model_save_path: null
    nclass: 2
    node_encoder_dim: 2
    num_layers: 2
    num_of_classes: 15
    num_relations: 12
    pooling_ratio: 0.5
    pooling_type: sagpool
    pretext_model: mrgcn
    readout_type: mean
    temporal_type: lstm_attn
relation_extraction_settings:
  desc: null
  value:
    ACTOR_NAMES:
    - ego car
    - car
    - moto
    - bicycle
    - ped
    - lane
    - light
    - sign
    - road
task_type:
  desc: null
  value: null
training_config:
  desc: null
  value:
    batch_size: 1
    dataset_type: scenegraph
    downsample: false
    enable_pretext: true
    epochs: 500
    learning_rate: 0.0001
    n_fold: 1
    pretext_lambda: 0.1
    pretext_loss: torch.nn.MSELoss()
    scenegraph_data_format: carla
    scenegraph_dataset_type: real
    seed: 0
    split_ratio: 0.3
    task_type: sequence_classification
    test_step: 5
    weight_decay: 0.0005
wandb_config:
  desc: null
  value:
    entity: <zhoumaomao>
    project: <RS2G with KL>
yaml_path:
  desc: null
  value: