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

Fix bug in Collector with Graph data (#456)

Browse Source 
* Fix bug in Collector with Graph data
* Add comments in DataModule class and bug fix in collate
This commit is contained in:
Filippo Olivo
2025-02-20 13:49:01 +01:00
committed by Nicola Demo
parent dfd6d7b467
commit 9c9d4fe7e4
6 changed files with 254 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pina/trainer.py
View File

@@ -105,9 +105,9 @@ class Trainer(lightning.pytorch.Trainer):
# checking compilation and automatic batching
if compile is None or sys.platform == "win32":
compile = False
if automatic_batching is None:
automatic_batching = False
self.automatic_batching = automatic_batching if automatic_batching \
is not None else False
# set attributes
self.compile = compile
self.solver = solver
@@ -115,7 +115,7 @@ class Trainer(lightning.pytorch.Trainer):
self._move_to_device()
self.data_module = None
self._create_datamodule(train_size, test_size, val_size, predict_size,
batch_size, automatic_batching, pin_memory,
batch_size, automatic_batching, pin_memory,
num_workers)
# logging