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/solvers/solver.py
View File

@@ -94,7 +94,7 @@ class SolverInterface(pytorch_lightning.LightningModule, metaclass=ABCMeta):
pass
@abstractmethod
def training_step(self):
def training_step(self, batch, batch_idx):
pass
@abstractmethod