File size: 315 Bytes
95b1715
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
train:
  train_runner: fse_editor
  train_dis: True
  dis_train_start_step: 45000
  steps: 500000


model:
  method: fse_full
  device: "0"
  batch_size: 8


encoder_losses:
  l2: 1.0
  lpips: 0.8
  id: 0.1
  adv: 0.01


disc_losses:
  main:
    coef: 1
  r1:
    coef: 10


optimizers:
    ranger:
      lr: 0.0001