app.py
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ def score(input_img):
|
|
| 37 |
class Config:
|
| 38 |
im_size = 224
|
| 39 |
batch_size = 1
|
| 40 |
-
num_workers =
|
| 41 |
# }}}
|
| 42 |
# 03. Define Dataset {{{
|
| 43 |
class PetDataset(Dataset):
|
|
@@ -183,7 +183,6 @@ def score(input_img):
|
|
| 183 |
|
| 184 |
test_preds = []
|
| 185 |
test_preds_model = []
|
| 186 |
-
modelfiles = glob(Config.model_base_dir + Config_exp77.model_dir + Config.model_file_ext)
|
| 187 |
test_preds_fold = []
|
| 188 |
model = PetNet_exp77(model_name = 'beit_large_patch16_224', out_features = 1, inp_channels = 3, pretrained=False)
|
| 189 |
model.load_state_dict(torch.load('beit_large_patch16_224_fold1_half.pth', map_location=torch.device('cpu')))
|
|
|
|
| 37 |
class Config:
|
| 38 |
im_size = 224
|
| 39 |
batch_size = 1
|
| 40 |
+
num_workers = 0
|
| 41 |
# }}}
|
| 42 |
# 03. Define Dataset {{{
|
| 43 |
class PetDataset(Dataset):
|
|
|
|
| 183 |
|
| 184 |
test_preds = []
|
| 185 |
test_preds_model = []
|
|
|
|
| 186 |
test_preds_fold = []
|
| 187 |
model = PetNet_exp77(model_name = 'beit_large_patch16_224', out_features = 1, inp_channels = 3, pretrained=False)
|
| 188 |
model.load_state_dict(torch.load('beit_large_patch16_224_fold1_half.pth', map_location=torch.device('cpu')))
|