!10217 Fix TextCNN training steps problem

From: @penny369
Reviewed-by: @oacjiewen,@linqingke
Signed-off-by: @linqingke
pull/10217/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 735820261a

@ -202,7 +202,6 @@ class MovieReview:
dataset = ds.GeneratorDataset(source=Generator(input_list=self.train), dataset = ds.GeneratorDataset(source=Generator(input_list=self.train),
column_names=["data", "label"], shuffle=False) column_names=["data", "label"], shuffle=False)
dataset = dataset.batch(batch_size=batch_size, drop_remainder=True) dataset = dataset.batch(batch_size=batch_size, drop_remainder=True)
dataset = dataset.repeat(epoch_size)
return dataset return dataset
def create_test_dataset(self, batch_size): def create_test_dataset(self, batch_size):

@ -90,7 +90,7 @@ class SoftmaxCrossEntropyExpand(Cell):
def make_conv_layer(kernel_size): def make_conv_layer(kernel_size):
return nn.Conv2d(in_channels=1, out_channels=96, kernel_size=kernel_size, padding=1, return nn.Conv2d(in_channels=1, out_channels=96, kernel_size=kernel_size, padding=1,
pad_mode="pad", weight_init='uniform', has_bias=True) pad_mode="pad", has_bias=True)
class TextCNN(nn.Cell): class TextCNN(nn.Cell):

Loading…
Cancel
Save