Commit ·
21ff76d
1
Parent(s): a18ead3
update
Browse files
utils.py
CHANGED
|
@@ -258,7 +258,8 @@ class MolecularGenerationModel():
|
|
| 258 |
|
| 259 |
@spaces.GPU(duration=60)
|
| 260 |
def generate(self, loader):
|
| 261 |
-
|
|
|
|
| 262 |
df = []
|
| 263 |
pbar = tqdm(loader, desc=f"Evaluating...", leave=False)
|
| 264 |
for it, batch in enumerate(pbar):
|
|
|
|
| 258 |
|
| 259 |
@spaces.GPU(duration=60)
|
| 260 |
def generate(self, loader):
|
| 261 |
+
self.model.to("cuda")
|
| 262 |
+
self.model.eval()
|
| 263 |
df = []
|
| 264 |
pbar = tqdm(loader, desc=f"Evaluating...", leave=False)
|
| 265 |
for it, batch in enumerate(pbar):
|