Skip to content

Commit 9d7a9fd

Browse files
authored
Fix beit base yaml (#131)
* add swin transformer * add swin transformer * add swin transformer * modify beit config
1 parent bd94e75 commit 9d7a9fd

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

configs/beit/beit_base_p16_224.yaml

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
epochs: 300
22
output_dir: output_dir
3+
seed: 0
4+
device: gpu
35

46
model:
5-
name: BeitWrapper
7+
name: BEiTWrapper
68
architecture:
79
name: Beit
810
img_size: 224
@@ -14,14 +16,16 @@ model:
1416
use_abs_pos_emb: False
1517
use_rel_pos_bias: True
1618
init_values: 0.1
17-
head:
18-
name: BeitClsHead
19+
head:
20+
name: BEiTClsHead
1921
num_classes: 1000
2022
in_channels: 768
2123

2224
dataloader:
2325
train:
24-
num_workers: 8
26+
loader:
27+
num_workers: 8
28+
use_shared_memory: True
2529
sampler:
2630
batch_size: 128
2731
shuffle: true

0 commit comments

Comments
 (0)