Tim77777767 commited on
Commit
001b062
·
1 Parent(s): e94b905

Anpassung config

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "model_type": "my_segformer",
3
  "in_channels": 3,
4
- "embed_dims": 32,
5
  "num_stages": 4,
6
  "num_layers": [2, 2, 2, 2],
7
  "num_heads": [1, 2, 5, 8],
 
1
  {
2
  "model_type": "my_segformer",
3
  "in_channels": 3,
4
+ "embed_dims": 64,
5
  "num_stages": 4,
6
  "num_layers": [2, 2, 2, 2],
7
  "num_heads": [1, 2, 5, 8],