Update README.md
Browse files
README.md
CHANGED
@@ -67,7 +67,7 @@ model = vggkagn(3,
|
|
67 |
affine=True
|
68 |
)
|
69 |
|
70 |
-
model.from_pretrained('
|
71 |
```
|
72 |
|
73 |
Transforms, used for validation on Imagenet1k:
|
|
|
67 |
affine=True
|
68 |
)
|
69 |
|
70 |
+
model.from_pretrained('brivangl/vgg_kagn11_v2')
|
71 |
```
|
72 |
|
73 |
Transforms, used for validation on Imagenet1k:
|