Skip to content

GitLab

Commit a97f5df7 authored by Kathy Wu's avatar Kathy Wu
Browse files

lint/doc fixes

parent 1f6b3d7d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/resnet/cifar10_main.py
View file @ a97f5df7
...@@ -142,6 +142,8 @@ def input_fn(is_training, data_dir, batch_size, num_epochs=1): ...@@ -142,6 +142,8 @@ def input_fn(is_training, data_dir, batch_size, num_epochs=1):
Args: Args:
is_training: A boolean denoting whether the input is for training. is_training: A boolean denoting whether the input is for training.
data_dir: The directory containing the input data.
batch_size: The number samples per batch.
num_epochs: The number of epochs to repeat the dataset. num_epochs: The number of epochs to repeat the dataset.
Returns: Returns:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment