ValueError: The `batch_size` arg

删除batch_size参数：

model.fit(dataset_train, batch_size=batch_size)

代码为：

model.fit(dataset_train)