folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve efficiency and refact LabelTensor, codacy correction and fix bug in PinaBatch

Browse Source
This commit is contained in:
FilippoOlivo
2024-10-23 15:04:28 +02:00
committed by Nicola Demo
parent ccc5f5a322
commit ea3d1924e7
13 changed files with 496 additions and 395 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pina/trainer.py
View File

@@ -79,7 +79,7 @@ class Trainer(pytorch_lightning.Trainer):
data_module = PinaDataModule(problem=self.solver.problem, device=device,
train_size=self.train_size,
test_size=self.test_size,
eval_size=self.eval_size)
val_size=self.eval_size)
data_module.setup()
self._loader = data_module.train_dataloader()