configs/config.yaml

train:
  epochs: 20
  batch_size: 16
  lr: 1.0E-5
  dropout: 0.1
  data_cut: null
  early_stop_count: 5

wandb:
  wandb_log: True
  wandb_project: 'paragraph_classification'
  wandb_group: 'model_test'
  wandb_memo: 'scibert'
  wandb_name: 'scibert'

model:
  model_name: 'allenai/scibert_scivocab_cased'
  data_file: './datasets/mix_dataset_2022_08_30.csv'
  max_length: 512
  random_state: 1000
  task_type: 'scalar'
  freeze_layers: null
  num_classifier: 1
  num_pos_emb_layer: 1
  sentence_piece: False
  bertsum: False

extract:
  selected_model: 'configs/best_model.pt'
  batch_size: 16