Update model.py
Browse files
model.py
CHANGED
|
@@ -2,7 +2,7 @@ import torchvision
|
|
| 2 |
import torch
|
| 3 |
from torch import nn
|
| 4 |
|
| 5 |
-
def vit_model(num_classes
|
| 6 |
|
| 7 |
|
| 8 |
|
|
@@ -22,7 +22,6 @@ def vit_model(num_classes,seed):
|
|
| 22 |
for params in model.parameters():
|
| 23 |
params.requires_grad = False
|
| 24 |
|
| 25 |
-
torch.manual_seed(seed)
|
| 26 |
|
| 27 |
# Change the output or classifier layer
|
| 28 |
model.heads = nn.Sequential(
|
|
|
|
| 2 |
import torch
|
| 3 |
from torch import nn
|
| 4 |
|
| 5 |
+
def vit_model(num_classes):
|
| 6 |
|
| 7 |
|
| 8 |
|
|
|
|
| 22 |
for params in model.parameters():
|
| 23 |
params.requires_grad = False
|
| 24 |
|
|
|
|
| 25 |
|
| 26 |
# Change the output or classifier layer
|
| 27 |
model.heads = nn.Sequential(
|