This commit is contained in:
Gabriel Ilharco Magalhaes 2018-02-20 11:27:11 -08:00
Родитель 4017a1664a
Коммит 84934eda67
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -328,9 +328,9 @@
" labelStream = C.io.StreamDef(field='labels', shape=num_label_classes, is_sparse=False)\n", " labelStream = C.io.StreamDef(field='labels', shape=num_label_classes, is_sparse=False)\n",
" featureStream = C.io.StreamDef(field='features', shape=input_dim, is_sparse=False)\n", " featureStream = C.io.StreamDef(field='features', shape=input_dim, is_sparse=False)\n",
" \n", " \n",
" deserailizer = C.io.CTFDeserializer(path, C.io.StreamDefs(labels = labelStream, features = featureStream))\n", " deserializer = C.io.CTFDeserializer(path, C.io.StreamDefs(labels = labelStream, features = featureStream))\n",
" \n", " \n",
" return C.io.MinibatchSource(deserailizer,\n", " return C.io.MinibatchSource(deserializer,\n",
" randomize = is_training, max_sweeps = C.io.INFINITELY_REPEAT if is_training else 1)" " randomize = is_training, max_sweeps = C.io.INFINITELY_REPEAT if is_training else 1)"
] ]
}, },