add attribute 'test' in dict 'data' (#264)

* add attribute 'test' in dict 'data'

* Update lenet5_mnist.py

 delete spaces around '='.

* init

* Delete custom directory
pull/282/head
Mingqiang Ning 2021-05-30 21:19:27 -05:00 committed by GitHub
parent 4524327a4a
commit 9a66113d20
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions

View File

@ -29,6 +29,8 @@ data = dict(
train=dict(
type=dataset_type, data_prefix='data/mnist', pipeline=train_pipeline),
val=dict(
type=dataset_type, data_prefix='data/mnist', pipeline=test_pipeline),
test=dict(
type=dataset_type, data_prefix='data/mnist', pipeline=test_pipeline))
evaluation = dict(
interval=5, metric='accuracy', metric_options={'topk': (1, )})