Manu8
/

vit_cifar

Image Classification

Generated from Trainer

Model card Files Files and versions

Metrics Training metrics Community

Manu8 commited on Jan 5, 2024

Commit

9ce273d

·

1 Parent(s): dc4f21b

Update model.py

Files changed (1) hide show

model.py +1 -1

model.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from .config import CNNConfig
 from transformers import PreTrainedModel
 import torch
 from torch import nn

+from .config import VITConfig
 from transformers import PreTrainedModel
 import torch
 from torch import nn