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"]
|