Update README.md
Browse files
README.md
CHANGED
|
@@ -41,7 +41,6 @@ bool_masked_pos = torch.randint(low=0, high=2, size=(1, num_patches)).bool()
|
|
| 41 |
|
| 42 |
outputs = model(pixel_values, bool_masked_pos=bool_masked_pos)
|
| 43 |
loss, logits = outputs.loss, outputs.logits
|
| 44 |
-
list(logits.shape)
|
| 45 |
```
|
| 46 |
|
| 47 |
### BibTeX entry and citation info
|
|
|
|
| 41 |
|
| 42 |
outputs = model(pixel_values, bool_masked_pos=bool_masked_pos)
|
| 43 |
loss, logits = outputs.loss, outputs.logits
|
|
|
|
| 44 |
```
|
| 45 |
|
| 46 |
### BibTeX entry and citation info
|