File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed
examples/data_efficient_nopt/config Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change 27
27
wget https://dataset.bj.bcebos.com/PaddleScience/data_efficient_nopt/possion_data/poisson_64_e5_15_val.h5
28
28
wget https://dataset.bj.bcebos.com/PaddleScience/data_efficient_nopt/possion_data/poisson_64_e5_15_test.h5
29
29
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
30
31
31
32
32
33
# pretrain
40
41
--config-name data_efficient_nopt_fno_poisson \
41
42
mode=finetune \
42
43
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"
44
45
```
45
46
46
47
=== "模型推理命令"
54
55
python data_efficient_nopt.py \
55
56
--config-name=data_efficient_nopt_fno_poisson.yaml \
56
57
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
58
59
```
59
60
60
61
## 1. 背景简介
Original file line number Diff line number Diff line change @@ -287,6 +287,7 @@ infer_config:
287
287
train_path : ' data/possion_64/poisson_64_e15_50_train.h5' # pick demos
288
288
test_path : ' data/possion_64/poisson_64_e15_50_test.h5'
289
289
scales_path : ' data/possion_64/poisson_64_e5_15_train_scale.npy'
290
+ ckpt_path : checkpoint/finetune_b01_m0_n8192.pdparams
290
291
291
292
num_data_workers : 0
292
293
subsample : 1
You can’t perform that action at this time.
0 commit comments