From 5a879eb2a2b347d48d4f3c04ca18ce4f73464ede Mon Sep 17 00:00:00 2001
From: Srijeet Roy <srijeet.11@gmail.com>
Date: Mon, 31 Jul 2023 16:30:45 +0200
Subject: [PATCH] update main.py
---
main.py | 12 +-----------
1 file changed, 1 insertion(+), 11 deletions(-)
diff --git a/main.py b/main.py
index e24ad8e..24c4277 100644
--- a/main.py
+++ b/main.py
@@ -115,16 +115,6 @@ def evaluate_func(f):
with open(f+"/dataset_setting.json","r") as fp:
dataset_setting = json.load(fp)
- # init dataloaders
- batchsize = meta_setting["batchsize"]
- test_dataset = globals()[meta_setting["dataset"]](train = False,**dataset_setting)
- #test_dataloader = torch.utils.data.DataLoader(test_dataset,batch_size=len(test_dataset), shuffle=False)
- test_dataloader = torch.utils.data.DataLoader(test_dataset,batch_size=batchsize, shuffle=False)
- # init UNet
- net = globals()[meta_setting["modelname"]](**model_setting).to(device)
- net = net.to(device)
- # init Diffusion Model
- framework = globals()[meta_setting["framework"]](net = net,device=device, **framework_setting)
print(f"META SETTINGS:\n\n {meta_setting}\n\n")
print(f"DATASET SETTINGS:\n\n {dataset_setting}\n\n")
@@ -133,7 +123,7 @@ def evaluate_func(f):
print(f"EVALUATION SETTINGS:\n\n {evaluation_setting}\n\n")
print("\n\nSTART EVALUATION\n\n")
- globals()[meta_setting["evaluation_function"]](model=framework, device=device, dataloader = test_dataloader,safepath = f,**evaluation_setting,)
+ globals()[meta_setting["evaluation_function"]](**evaluation_setting)
print("\n\nFINISHED EVALUATION\n\n")
--
GitLab