AttributeError: 'FIDModel' object has no attribute 'add_sample'

[LokeshBadisa]

I installed pytorch-fid from https://github.com/zengxianyu/pytorch-fid. It still gives me AttributeError: 'FIDModel' object has no attribute 'add_sample'. There is no add_sample function.

fid_model.py has:

class FIDModel(nn.Module):
    def __init__(self):
        super(FIDModel, self).__init__()
        self.model = InceptionV3()

    def forward(self, x):
        return self.model(x)