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
|