diff --git a/examples/eval_pretrained.py b/examples/eval_pretrained.py index 4cdb6db..96d66df 100644 --- a/examples/eval_pretrained.py +++ b/examples/eval_pretrained.py @@ -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 @@ -45,7 +45,7 @@ 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 @@ -53,6 +53,7 @@ name to give your file. """ + class CustomDataset(torch.utils.data.Dataset): def __init__(self): super().__init__()