search

PaddlePaddle / PaddleScience

PaddleScience is SDK and library for developing AI-driven scientific computing applications based on PaddlePaddle.
http://paddlescience-docs.rtfd.io/
Apache License 2.0
244 stars 132 forks source link

[Refine] Refine loss and metric module #919

Closed HydrogenSulfate closed 3 weeks ago

HydrogenSulfate commented 1 month ago

PR types

Function optimization

PR changes

APIs

Describe

  1. 修改 loss 模块的返回值，从标量Tensor修改为未经过求和的Dict[str, Tensor]，使得aggregator模块接收到的losses是output_keys粒度而不是constraint_name粒度的字典，以应对同一个constraint不同output_key具有不同的权重的情况。
  2. 修改所有使用了loss模块的代码和Examples样例
  3. 删除Constraint中将area添加到output_keys的逻辑
paddle-bot[bot] commented 1 month ago

Thanks for your contribution!