Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
3c0cedc
fix(test=document_fix)
HydrogenSulfate Nov 23, 2023
ebb31a1
Merge branch 'develop' of https://github.com/HydrogenSulfate/PaddleSc…
HydrogenSulfate Nov 23, 2023
21c216c
Merge branch 'develop' of https://github.com/HydrogenSulfate/PaddleSc…
HydrogenSulfate Nov 24, 2023
01bf607
Merge branch 'develop' of https://github.com/HydrogenSulfate/PaddleSc…
HydrogenSulfate Nov 24, 2023
a1dfb17
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate Nov 28, 2023
865de8d
Merge branches 'develop' and 'develop' of github.com:HydrogenSulfate/…
HydrogenSulfate Dec 2, 2023
70324ae
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate Dec 8, 2023
1a22676
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate Dec 11, 2023
a345539
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate Dec 14, 2023
22feda0
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate Dec 29, 2023
fa3ec47
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate Jan 2, 2024
b4c2087
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate Jan 8, 2024
264da4b
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate Jan 19, 2024
93c3edd
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate Jan 26, 2024
f5edb52
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate Feb 27, 2024
4e580b9
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate Feb 27, 2024
8f9ae46
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate Mar 13, 2024
e8570a1
Merge branch 'develop' of github.com:HydrogenSulfate/PaddleScience in…
HydrogenSulfate May 6, 2024
2d69a73
fix typo for correcting 'pdpiparmas' to 'pdiparams'
HydrogenSulfate May 6, 2024
1c86107
add left bracket for docstring of CuboidSelfAttentionLayer
HydrogenSulfate May 6, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion examples/NLS-MB/conf/NLS-MB_rogue_wave.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@ INFER:
pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/NLS-MB/NLS-MB_rogue_wave_pretrained.pdparams
export_path: ./inference/NLS-MB_rogue_wave
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
device: gpu
engine: native
precision: fp32
Expand Down
2 changes: 1 addition & 1 deletion examples/NLS-MB/conf/NLS-MB_soliton.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@ INFER:
pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/NLS-MB/NLS-MB_soliton_pretrained.pdparams
export_path: ./inference/NLS-MB_soliton
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
device: gpu
engine: native
precision: fp32
Expand Down
2 changes: 1 addition & 1 deletion examples/allen_cahn/conf/allen_cahn.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ INFER:
pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/AllenCahn/allen_cahn_plain_pretrained.pdparams
export_path: ./inference/allen_cahn
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
onnx_path: ${INFER.export_path}.onnx
device: gpu
engine: native
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ INFER:
pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/AllenCahn/allen_cahn_plain_pretrained.pdparams
export_path: ./inference/allen_cahn
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
onnx_path: ${INFER.export_path}.onnx
device: gpu
engine: native
Expand Down
2 changes: 1 addition & 1 deletion examples/allen_cahn/conf/allen_cahn_default.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ INFER:
pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/AllenCahn/allen_cahn_plain_pretrained.pdparams
export_path: ./inference/allen_cahn
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
onnx_path: ${INFER.export_path}.onnx
device: gpu
engine: native
Expand Down
2 changes: 1 addition & 1 deletion examples/aneurysm/conf/aneurysm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,7 @@ INFER:
pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/aneurysm/aneurysm_pretrained.pdparams
export_path: ./inference/aneurysm
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
onnx_path: ${INFER.export_path}.onnx
device: gpu
engine: native
Expand Down
2 changes: 1 addition & 1 deletion examples/biharmonic2d/conf/biharmonic2d.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ INFER:
pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/biharmonic2d/biharmonic2d_pretrained.pdparams
export_path: ./inference/biharmonic2d
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
output_keys: ["Mx", "Mxy", "My", "Qx", "Qy", "w"]
device: gpu
engine: native
Expand Down
2 changes: 1 addition & 1 deletion examples/earthformer/conf/earthformer_enso_pretrain.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -133,7 +133,7 @@ INFER:
pretrained_model_path: ./checkpoint/enso/earthformer_enso.pdparams
export_path: ./inference/earthformer/enso
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
device: gpu
engine: native
precision: fp32
Expand Down
2 changes: 1 addition & 1 deletion examples/earthformer/conf/earthformer_sevir_pretrain.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -161,7 +161,7 @@ INFER:
pretrained_model_path: ./checkpoint/sevir/earthformer_sevir.pdparams
export_path: ./inference/earthformer/sevir
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
device: gpu
engine: native
precision: fp32
Expand Down
2 changes: 1 addition & 1 deletion examples/earthformer/predictor.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ def __init__(
):
super().__init__(
cfg.INFER.pdmodel_path,
cfg.INFER.pdpiparams_path,
cfg.INFER.pdiparams_path,
device=cfg.INFER.device,
engine=cfg.INFER.engine,
precision=cfg.INFER.precision,
Expand Down
2 changes: 1 addition & 1 deletion examples/euler_beam/conf/euler_beam.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ INFER:
pretrained_model_path: "https://paddle-org.bj.bcebos.com/paddlescience/models/euler_beam/euler_beam_pretrained.pdparams"
export_path: ./inference/euler_beam
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
device: gpu
engine: native
precision: fp32
Expand Down
2 changes: 1 addition & 1 deletion examples/fsi/conf/viv.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ INFER:
pretrained_model_path: "https://paddle-org.bj.bcebos.com/paddlescience/models/viv/viv_pretrained.pdparams"
export_path: ./inference/viv
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
input_keys: ${MODEL.input_keys}
output_keys: ["eta", "f"]
device: gpu
Expand Down
2 changes: 1 addition & 1 deletion examples/ide/conf/volterra_ide.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ INFER:
pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/volterra_ide/volterra_ide_pretrained.pdparams
export_path: ./inference/volterra_ide
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
device: gpu
engine: native
precision: fp32
Expand Down
2 changes: 1 addition & 1 deletion examples/laplace/conf/laplace2d.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ INFER:
pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/laplace2d/laplace2d_pretrained.pdparams
export_path: ./inference/laplace2d
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
device: gpu
engine: native
precision: fp32
Expand Down
2 changes: 1 addition & 1 deletion examples/lorenz/conf/transformer.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ INFER:
pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/lorenz/lorenz_transformer_pretrained.pdparams
export_path: ./inference/lorenz_transformer
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
device: gpu
engine: native
precision: fp32
Expand Down
2 changes: 1 addition & 1 deletion examples/nsfnet/conf/VP_NSFNet4.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ INFER:
pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/nsfnet/nsfnet4.pdparams
export_path: ./inference/VP_NSFNet4
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
output_keys: ['p', 'u', 'v', 'w']
device: gpu
engine: native
Expand Down
2 changes: 1 addition & 1 deletion examples/rossler/conf/transformer.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ INFER:
pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/rossler/rossler_transformer_pretrained.pdparams
export_path: ./inference/rossler_transformer
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
device: gpu
engine: native
precision: fp32
Expand Down
2 changes: 1 addition & 1 deletion examples/topopt/conf/topopt.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ INFER:
pretrained_model_path_dict: {'Uniform': 'https://paddle-org.bj.bcebos.com/paddlescience/models/topopt/uniform_pretrained.pdparams', 'Poisson5': 'https://paddle-org.bj.bcebos.com/paddlescience/models/topopt/poisson5_pretrained.pdparams', 'Poisson10': 'https://paddle-org.bj.bcebos.com/paddlescience/models/topopt/poisson10_pretrained.pdparams', 'Poisson30': 'https://paddle-org.bj.bcebos.com/paddlescience/models/topopt/poisson30_pretrained.pdparams'}
export_path: ./inference/topopt_${INFER.pretrained_model_name}
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
device: gpu
engine: native
precision: fp32
Expand Down
2 changes: 1 addition & 1 deletion examples/yinglong1/conf/yinglong_12.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ INFER:
pretrained_model_path: null
export_path: ./inference/yinglong_12
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
onnx_path: ${INFER.export_path}.onnx
device: gpu
engine: native
Expand Down
2 changes: 1 addition & 1 deletion examples/yinglong1/conf/yinglong_24.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ INFER:
pretrained_model_path: null
export_path: ./inference/yinglong_24
pdmodel_path: ${INFER.export_path}.pdmodel
pdpiparams_path: ${INFER.export_path}.pdiparams
pdiparams_path: ${INFER.export_path}.pdiparams
onnx_path: ${INFER.export_path}.onnx
device: gpu
engine: native
Expand Down
2 changes: 1 addition & 1 deletion examples/yinglong1/predictor.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def __init__(
):
super().__init__(
cfg.INFER.pdmodel_path,
cfg.INFER.pdpiparams_path,
cfg.INFER.pdiparams_path,
device=cfg.INFER.device,
engine=cfg.INFER.engine,
precision=cfg.INFER.precision,
Expand Down
4 changes: 2 additions & 2 deletions ppsci/arch/cuboid_transformer_encoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -460,10 +460,10 @@ class CuboidSelfAttentionLayer(paddle.nn.Layer):

We adopt two mechanisms for decomposing the input tensor into cuboids:

1) local:
(1) local:
We group the tensors within a local window, e.g., X[t:(t+b_t), h:(h+b_h), w:(w+b_w)]. We can also apply the
shifted window strategy proposed in "[ICCV2021] Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
2) dilated:
(2) dilated:
Inspired by the success of dilated convolution "[ICLR2016] Multi-Scale Context Aggregation by Dilated Convolutions",
we split the tensor with dilation factors that are tied to the size of the cuboid. For example, for a cuboid that has width `b_w`,
we sample the elements starting from 0 as 0, w / b_w, 2 * w / b_w, ..., (b_w - 1) * w / b_w.
Expand Down
2 changes: 1 addition & 1 deletion ppsci/utils/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -197,7 +197,7 @@ class InferConfig(BaseModel):
pretrained_model_path: Optional[str] = None
export_path: str
pdmodel_path: Optional[str] = None
pdpiparams_path: Optional[str] = None
pdiparams_path: Optional[str] = None
onnx_path: Optional[str] = None
device: Literal["gpu", "cpu", "npu", "xpu"] = "cpu"
engine: Literal["native", "tensorrt", "onnx", "mkldnn"] = "native"
Expand Down