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
|