Skip to content

Commit

Permalink
black
Browse files Browse the repository at this point in the history
  • Loading branch information
eatpk committed Feb 19, 2024
1 parent 0bdd56b commit ae01df6
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions examples/mnist_influence/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ def construct_mlp(num_inputs=784, num_classes=10, seed=0):
)
return model


def get_mnist_dataloader(
batch_size=128,
split="train",
Expand Down Expand Up @@ -64,6 +65,7 @@ def get_mnist_dataloader(
drop_last=drop_last,
)


def get_fmnist_dataloader(
batch_size=128,
split="train",
Expand Down

0 comments on commit ae01df6

Please sign in to comment.