yeq6x commited on
Commit
d78ce10
·
1 Parent(s): 0c6ebb1

並び替え

Browse files
Files changed (1) hide show
  1. app.py +6 -4
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 = [{"name": "ae_model_tf_2024-03-05_00-35-21.pth", "feature_dim": 32},
76
- {"name": "autoencoder-epoch=09-train_loss=1.00.ckpt", "feature_dim": 64},
77
- {"name": "autoencoder-epoch=29-train_loss=1.01.ckpt", "feature_dim": 64},
78
- {"name": "autoencoder-epoch=49-train_loss=1.01.ckpt", "feature_dim": 64}]
 
 
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"]