Skip to content

Commit

Permalink
space formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
kwotsin committed Aug 11, 2020
1 parent 1a54fb7 commit fd3c512
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions examples/eval_pretrained.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from torch_mimicry.nets import sngan

######################################################
# Computing Metrics with Default Datasets
# Computing Metrics with Default Datasets
######################################################

# Download cifar10 checkpoint: https://drive.google.com/uc?id=1Gn4ouslRAHq3D7AP_V-T2x8Wi1S1hTXJ&export=download
Expand Down Expand Up @@ -45,14 +45,15 @@
device=device)

######################################################
# Computing Metrics with Custom Datasets
# Computing Metrics with Custom Datasets
######################################################
"""
Simply define a custom dataset as below to compute FID/KID, and define
a stats_file/feat_file to save the cached statistics since we don't know what
name to give your file.
"""


class CustomDataset(torch.utils.data.Dataset):
def __init__(self):
super().__init__()
Expand Down

0 comments on commit fd3c512

Please sign in to comment.