Commit ·
fd1d2e7
1
Parent(s): 0c717d3
Add application file with new chkpoint
Browse files- .DS_Store +0 -0
- app.py +1 -1
- logs/.DS_Store +0 -0
- logs/checkpoints/{epoch=58-val_loss=1.46.ckpt → epoch=59-val_loss=1.46.ckpt} +1 -1
.DS_Store
CHANGED
|
Binary files a/.DS_Store and b/.DS_Store differ
|
|
|
app.py
CHANGED
|
@@ -186,7 +186,7 @@ class_names_2 = list(class_names_dict.keys())
|
|
| 186 |
def load_model():
|
| 187 |
# Load the trained model
|
| 188 |
model = ImageNetClassifier.load_from_checkpoint(
|
| 189 |
-
"logs/checkpoints/epoch=
|
| 190 |
map_location="cuda" if torch.cuda.is_available() else "cpu",
|
| 191 |
#lr=6.28E-02, # This parameter is required but won't be used for inference
|
| 192 |
num_classes=1000 # Make sure this matches your trained model
|
|
|
|
| 186 |
def load_model():
|
| 187 |
# Load the trained model
|
| 188 |
model = ImageNetClassifier.load_from_checkpoint(
|
| 189 |
+
"logs/checkpoints/epoch=59-val_loss=1.46.ckpt",
|
| 190 |
map_location="cuda" if torch.cuda.is_available() else "cpu",
|
| 191 |
#lr=6.28E-02, # This parameter is required but won't be used for inference
|
| 192 |
num_classes=1000 # Make sure this matches your trained model
|
logs/.DS_Store
CHANGED
|
Binary files a/logs/.DS_Store and b/logs/.DS_Store differ
|
|
|
logs/checkpoints/{epoch=58-val_loss=1.46.ckpt → epoch=59-val_loss=1.46.ckpt}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 307147814
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d4533925de5f0c0d6ad5a5c116d3a6d7b62d8b03eb6fce450850018eea1732c6
|
| 3 |
size 307147814
|