Commit
•
c7179a0
1
Parent(s):
28f1ee4
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,6 +6,7 @@ import torch.nn as nn
|
|
| 6 |
import os
|
| 7 |
|
| 8 |
model_name = "dinov2-large-2024_01_24-with_data_aug_batch-size32_epochs93_freeze"
|
|
|
|
| 9 |
|
| 10 |
# CREATE CUSTOM MODEL
|
| 11 |
def create_head(num_features , number_classes ,dropout_prob=0.5 ,activation_func =nn.ReLU):
|
|
|
|
| 6 |
import os
|
| 7 |
|
| 8 |
model_name = "dinov2-large-2024_01_24-with_data_aug_batch-size32_epochs93_freeze"
|
| 9 |
+
checkpoint_name = "lombardata/" + model_name
|
| 10 |
|
| 11 |
# CREATE CUSTOM MODEL
|
| 12 |
def create_head(num_features , number_classes ,dropout_prob=0.5 ,activation_func =nn.ReLU):
|