Update README.md
Browse files
README.md
CHANGED
|
@@ -63,8 +63,8 @@ print(emb.shape) # torch.Size([2, 1024])
|
|
| 63 |
```
|
| 64 |
|
| 65 |
`layer` can be
|
|
|
|
| 66 |
- `"first_fc"` (1024-d, the default and recommended),
|
| 67 |
-
- `"GNN"` (75-d), or
|
| 68 |
- `"second_fc"` (128-d).
|
| 69 |
|
| 70 |
## Training data
|
|
|
|
| 63 |
```
|
| 64 |
|
| 65 |
`layer` can be
|
| 66 |
+
- `"GNN"` (75-d),
|
| 67 |
- `"first_fc"` (1024-d, the default and recommended),
|
|
|
|
| 68 |
- `"second_fc"` (128-d).
|
| 69 |
|
| 70 |
## Training data
|