search

ehsanhaghighat / sciann

Deep learning for Engineers - Physics Informed Deep Learning
htttp://sciann.com
Other
325 stars 116 forks source link

Custom loss weight for SciMmodel #102

Open tirtho109 opened 7 months ago

tirtho109 commented 7 months ago

https://github.com/sciann/sciann/blob/e3615412c149dbf3152433c09cdd741be2b04f62/sciann/models/model.py#L139

Here, all loss_weights for individual loss term are set to 1.0 by default. How can I set custom weights to loss terms of SciModel object before training.

Any short example would help. Thank you!