Fix bugs (#387)
This commit is contained in:
committed by
Nicola Demo
parent
3c95441aac
commit
f748b66194
@@ -134,16 +134,18 @@ class SolverInterface(lightning.pytorch.LightningModule, metaclass=ABCMeta):
|
||||
return super().on_train_start()
|
||||
|
||||
def _check_solver_consistency(self, problem):
|
||||
"""
|
||||
TODO
|
||||
"""
|
||||
pass
|
||||
#TODO : Implement this method for the conditions
|
||||
'''
|
||||
|
||||
|
||||
for _, condition in problem.conditions.items():
|
||||
if not set(condition.condition_type).issubset(
|
||||
set(self.accepted_condition_types)):
|
||||
raise ValueError(
|
||||
f'{self.__name__} dose not support condition '
|
||||
f'{condition.condition_type}')
|
||||
|
||||
'''
|
||||
@staticmethod
|
||||
def get_batch_size(batch):
|
||||
# Assuming batch is your custom Batch object
|
||||
|
||||
Reference in New Issue
Block a user