Update solvers (#434)
* Enable DDP training with batch_size=None and add validity check for split sizes * Refactoring SolverInterfaces (#435) * Solver update + weighting * Updating PINN for 0.2 * Modify GAROM + tests * Adding more versatile loggers * Disable compilation when running on Windows * Fix tests --------- Co-authored-by: giovanni <giovanni.canali98@yahoo.it> Co-authored-by: FilippoOlivo <filippo@filippoolivo.com>
This commit is contained in:
committed by
Nicola Demo
parent
780c4921eb
commit
9cae9a438f
@@ -1,5 +1,4 @@
|
||||
from . import LabelTensor
|
||||
from .utils import check_consistency, merge_tensors
|
||||
from .utils import check_consistency
|
||||
|
||||
|
||||
class Collector:
|
||||
@@ -8,11 +7,6 @@ class Collector:
|
||||
# creating a hook between collector and problem
|
||||
self.problem = problem
|
||||
|
||||
# this variable is used to store the data in the form:
|
||||
# {'[condition_name]' :
|
||||
# {'input_points' : Tensor,
|
||||
# '[equation/output_points/conditional_variables]': Tensor}
|
||||
# }
|
||||
# those variables are used for the dataloading
|
||||
self._data_collections = {name: {} for name in self.problem.conditions}
|
||||
self.conditions_name = {
|
||||
|
||||
Reference in New Issue
Block a user