Update README.md
Browse files
README.md
CHANGED
|
@@ -31,4 +31,16 @@ filename = "vit_large_patch14_224_skyclip_50pct.pth"
|
|
| 31 |
torch.save(model_timm.state_dict(), filename)
|
| 32 |
md5 = hashlib.md5(open(filename, "rb").read()).hexdigest()[:8]
|
| 33 |
os.rename(filename, filename.replace(".pth", f"-{md5}.pth"))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
```
|
|
|
|
| 31 |
torch.save(model_timm.state_dict(), filename)
|
| 32 |
md5 = hashlib.md5(open(filename, "rb").read()).hexdigest()[:8]
|
| 33 |
os.rename(filename, filename.replace(".pth", f"-{md5}.pth"))
|
| 34 |
+
```
|
| 35 |
+
|
| 36 |
+
The preprocessing transforms are:
|
| 37 |
+
|
| 38 |
+
```
|
| 39 |
+
Compose(
|
| 40 |
+
Resize(size=224, interpolation=bicubic, max_size=None, antialias=True)
|
| 41 |
+
CenterCrop(size=(224, 224))
|
| 42 |
+
<function _convert_to_rgb at 0x707759392980>
|
| 43 |
+
ToTensor()
|
| 44 |
+
Normalize(mean=(0.48145466, 0.4578275, 0.40821073), std=(0.26862954, 0.26130258, 0.27577711))
|
| 45 |
+
)
|
| 46 |
```
|