initialize self-adaptive weights to 0 (#358)
This commit is contained in:
committed by
Nicola Demo
parent
801b6b8d34
commit
53605d5a09
@@ -117,7 +117,7 @@ class RBAPINN(PINN):
|
|||||||
# initialize weights
|
# initialize weights
|
||||||
self.weights = {}
|
self.weights = {}
|
||||||
for condition_name in problem.conditions:
|
for condition_name in problem.conditions:
|
||||||
self.weights[condition_name] = 1
|
self.weights[condition_name] = 0
|
||||||
|
|
||||||
# define vectorial loss
|
# define vectorial loss
|
||||||
self._vectorial_loss = deepcopy(loss)
|
self._vectorial_loss = deepcopy(loss)
|
||||||
|
|||||||
Reference in New Issue
Block a user