svhn2mnist.yaml

data_transforms:
  input_size: 32

trainer:
  n_step_f: 4 # MCD N_STEP_F
  batch_size: 128
#  max_epochs: 200
  num_workers: 4
  save_model_addr: 'checkpoints/svhn2mnist'
  forget_rate: 0.5

optimizer:
  name: 'SGD'
  params:
    lr: 0.001
    momentum: 0.9
    weight_decay: 0.0005

data:
  type: 'digits'
  class_number: 10
  root: 'data/digits_dg'
  source_domain_x: 'svhn'
  source_domain_u_1: 'mnist_m'
  source_domain_u_2: 'syn'
  target_domain: 'mnist'

log:
  save_addr: 'log'
  save_name: 'svhn2mnist'

ratio: 0.4 # a ratio to split samples into clean set and noisy set