Hugging Face's logo

Spaces:
swimmiing
/
ACL-SSL-zeroshot-demo
Running

App Files Community
ACL-SSL-zeroshot-demo
File size: 646 Bytes 
b20af9f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
 
model: ACL

common:
  train_data: vggss
  epoch: 20
  batch_size: 8
  input_resolution: 352
  num_workers: 4
  seed: 0
  loss:
    - acl_i
    - acl_f
    - area_reg
  loss_w:
    - 1
    - 1
    - 1

optimizer: Adam
scheduler: null
amp: True

optim_conf:
  Adam:
    module_path: torch.optim
    module_name: Adam
    lr: 0.0001
    weight_decay: 0.0001

  AdamW:
    module_path: torch.optim
    module_name: AdamW
    lr: 0.001

  SGDR:
    module_path: torch.optim
    module_name: SGD
    lr: 0.5
    weight_decay: 0.00001

sched_conf:
  Cosine:
    module_path: torch.optim.lr_scheduler
    module_name: CosineAnnealingLR
    eta_ratio: 0.0