# only needs torch and torchvision

if __name__ == '__main__':
    pass