AbstractPhil commited on
Commit
45bae27
·
verified ·
1 Parent(s): e37d8c5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -0
README.md CHANGED
@@ -5,6 +5,33 @@ license: mit
5
 
6
  The decoder is differentiating. The features will be useful downstream.
7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  ```
9
  Note: Environment variable`HF_TOKEN` is set and is the current active token independently from the token you've just configured.
10
  WARNING:huggingface_hub._login:Note: Environment variable`HF_TOKEN` is set and is the current active token independently from the token you've just configured.
 
5
 
6
  The decoder is differentiating. The features will be useful downstream.
7
 
8
+ ```python
9
+ CFG = dict(
10
+ # Architecture (inherited from Fresnel v50)
11
+ V=16, D=4, ps=4, hidden=384, depth=4, n_cross=2,
12
+ stage_hidden=128, stage_V=64,
13
+
14
+ # Training
15
+ img_size=64,
16
+ batch_size=256,
17
+ lr=3e-4,
18
+ epochs=50,
19
+ ds_size=1280000,
20
+ val_size=10000,
21
+
22
+ # CV soft hand
23
+ target_cv=0.2915,
24
+ cv_weight=0.3,
25
+ boost=0.5,
26
+ sigma=0.15,
27
+
28
+ # Checkpointing
29
+ save_every=5,
30
+ val_per_type_every=5,
31
+ )
32
+ ```
33
+
34
+
35
  ```
36
  Note: Environment variable`HF_TOKEN` is set and is the current active token independently from the token you've just configured.
37
  WARNING:huggingface_hub._login:Note: Environment variable`HF_TOKEN` is set and is the current active token independently from the token you've just configured.