Skip to content
This repository was archived by the owner on Jul 7, 2023. It is now read-only.

Commit 2d2d160

Browse files
Lukasz Kaisercopybara-github
Lukasz Kaiser
authored andcommitted
Correct flat CIFAR modality to not consider 0 as padding.
PiperOrigin-RevId: 240682373
1 parent 211c824 commit 2d2d160

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensor2tensor/data_generators/cifar.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ def hparams(self, defaults, model_hparams):
197197
super(ImageCifar10PlainGenFlat, self).hparams(defaults, model_hparams)
198198
# Switch to symbol modality
199199
p = defaults
200-
p.modality["inputs"] = modalities.ModalityType.SYMBOL
200+
p.modality["inputs"] = modalities.ModalityType.SYMBOL_WEIGHTS_ALL
201201
p.input_space_id = problem.SpaceID.GENERIC
202202

203203

0 commit comments

Comments
 (0)