add encoding to yaml loading and apply torch.no_grad to test

This commit is contained in:
2024-01-04 20:42:58 +08:00
parent 465b163a73
commit 9404557763
8 changed files with 42 additions and 23 deletions

View File

@ -5,8 +5,8 @@ stages:
deps:
- path: prepare.py
hash: md5
md5: a1c07d1d5caf6e5288560a189415785c
size: 2979
md5: f0fd9944ebe22e1af020f031847991ed
size: 2992
params:
params.yaml:
prepare:
@ -19,21 +19,21 @@ stages:
outs:
- path: data/processed
hash: md5
md5: f4bf62ffa725ca9144b7852a283dc1da.dir
size: 295118798
nfiles: 60000
md5: a47a61b2a1709f487fd286e9c54b89fc.dir
size: 295124946
nfiles: 60001
train:
cmd: python train.py
deps:
- path: data/processed
hash: md5
md5: f4bf62ffa725ca9144b7852a283dc1da.dir
size: 295118798
nfiles: 60000
md5: a47a61b2a1709f487fd286e9c54b89fc.dir
size: 295124946
nfiles: 60001
- path: train.py
hash: md5
md5: b797ccf2fe61952bbf6d83fa51b0b11f
size: 3407
md5: aabaf1a407badf48c97b14a69b0072ea
size: 3420
params:
params.yaml:
train:
@ -46,3 +46,23 @@ stages:
hash: md5
md5: 8ead2a7cd52d70b359d3cdc3df5e43e3
size: 102592994
evaluate:
cmd: python evaluate.py
deps:
- path: data/processed
hash: md5
md5: a47a61b2a1709f487fd286e9c54b89fc.dir
size: 295124946
nfiles: 60001
- path: evaluate.py
hash: md5
md5: 8a9a2e95a6b64e632a4f2feac62d294b
size: 1473
- path: model.pt
hash: md5
md5: 8ead2a7cd52d70b359d3cdc3df5e43e3
size: 102592994
params:
params.yaml:
evaluate:
data_dir: data/processed