Skip to content

Commit 9930bcb

Browse files
committed
feat(ppsci): support data_effient_nopt
1 parent a57c461 commit 9930bcb

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

docs/zh/examples/data_efficient_nopt.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
wget https://dataset.bj.bcebos.com/PaddleScience/data_efficient_nopt/possion_data/poisson_64_e5_15_val.h5
2828
wget https://dataset.bj.bcebos.com/PaddleScience/data_efficient_nopt/possion_data/poisson_64_e5_15_test.h5
2929
wget https://dataset.bj.bcebos.com/PaddleScience/data_efficient_nopt/possion_data/train_rand_idx.npy
30+
wget https://dataset.bj.bcebos.com/PaddleScience/data_efficient_nopt/possion_data/pretrain_b01_m0.pdparams
3031

3132

3233
# pretrain
@@ -40,7 +41,7 @@
4041
--config-name data_efficient_nopt_fno_poisson \
4142
mode=finetune \
4243
config=pois_64_finetune_e5_15 \
43-
train_config.pois_64_finetune_e5_15.pretrained_ckpt_path="./data/pretrain_b01_m0.pdparams"
44+
train_config.pois_64_finetune_e5_15.pretrained_ckpt_path="./data/possion_64/pretrain_b01_m0.pdparams"
4445
```
4546

4647
=== "模型推理命令"
@@ -54,7 +55,7 @@
5455
python data_efficient_nopt.py \
5556
--config-name=data_efficient_nopt_fno_poisson.yaml \
5657
mode=infer \
57-
infer_config.ckpt_path=./exp/pois_64_finetune_e5_15/r0/training_checkpoints/ckpt.tar
58+
infer_config.ckpt_path=./data/possion_64/finetune_b01_m0_n8192.pdparams
5859
```
5960

6061
## 1. 背景简介

examples/data_efficient_nopt/config/data_efficient_nopt_fno_poisson.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -287,6 +287,7 @@ infer_config:
287287
train_path: 'data/possion_64/poisson_64_e15_50_train.h5' # pick demos
288288
test_path: 'data/possion_64/poisson_64_e15_50_test.h5'
289289
scales_path: 'data/possion_64/poisson_64_e5_15_train_scale.npy'
290+
ckpt_path: checkpoint/finetune_b01_m0_n8192.pdparams
290291

291292
num_data_workers: 0
292293
subsample: 1

0 commit comments

Comments
 (0)