Update README.md
Browse files
README.md
CHANGED
|
@@ -40,8 +40,6 @@ model.eval()
|
|
| 40 |
|
| 41 |
inp = torch.ones(1,3,224,224)
|
| 42 |
out = model(inp)
|
| 43 |
-
# print(out.logits)
|
| 44 |
-
print(torch.sum(out.logits))
|
| 45 |
```
|
| 46 |
|
| 47 |
## Citation
|
|
|
|
| 40 |
|
| 41 |
inp = torch.ones(1,3,224,224)
|
| 42 |
out = model(inp)
|
|
|
|
|
|
|
| 43 |
```
|
| 44 |
|
| 45 |
## Citation
|