Spaces:
Sleeping
Sleeping
並び替え
Browse files
app.py
CHANGED
|
@@ -72,10 +72,12 @@ image_size = 112
|
|
| 72 |
batch_size = 32
|
| 73 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 74 |
|
| 75 |
-
models_info = [
|
| 76 |
-
|
| 77 |
-
|
| 78 |
-
|
|
|
|
|
|
|
| 79 |
models = []
|
| 80 |
for model_info in models_info:
|
| 81 |
model_name = model_info["name"]
|
|
|
|
| 72 |
batch_size = 32
|
| 73 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 74 |
|
| 75 |
+
models_info = [
|
| 76 |
+
{"name": "autoencoder-epoch=49-train_loss=1.01.ckpt", "feature_dim": 64},
|
| 77 |
+
{"name": "autoencoder-epoch=29-train_loss=1.01.ckpt", "feature_dim": 64},
|
| 78 |
+
{"name": "autoencoder-epoch=09-train_loss=1.00.ckpt", "feature_dim": 64},
|
| 79 |
+
{"name": "ae_model_tf_2024-03-05_00-35-21.pth", "feature_dim": 32},
|
| 80 |
+
]
|
| 81 |
models = []
|
| 82 |
for model_info in models_info:
|
| 83 |
model_name = model_info["name"]
|