add encoding to yaml loading and apply torch.no_grad to test
This commit is contained in:
40
dvc.lock
40
dvc.lock
@ -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
|
||||
|
||||
Reference in New Issue
Block a user