We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bd94e75 commit 9d7a9fdCopy full SHA for 9d7a9fd
configs/beit/beit_base_p16_224.yaml
@@ -1,8 +1,10 @@
1
epochs: 300
2
output_dir: output_dir
3
+seed: 0
4
+device: gpu
5
6
model:
- name: BeitWrapper
7
+ name: BEiTWrapper
8
architecture:
9
name: Beit
10
img_size: 224
@@ -14,14 +16,16 @@ model:
14
16
use_abs_pos_emb: False
15
17
use_rel_pos_bias: True
18
init_values: 0.1
- head:
- name: BeitClsHead
19
+ head:
20
+ name: BEiTClsHead
21
num_classes: 1000
22
in_channels: 768
23
24
dataloader:
25
train:
- num_workers: 8
26
+ loader:
27
+ num_workers: 8
28
+ use_shared_memory: True
29
sampler:
30
batch_size: 128
31
shuffle: true
0 commit comments