black0017

from torch.optim.optimizer import Optimizer, required
import torch

# almost copy paste from https://github.com/noahgolmant/pytorch-lars/blob/master/lars.py
class LARS(Optimizer):
    r"""Implements LARS (Layer-wise Adaptive Rate Scaling).
    Args:
        params (iterable): iterable of parameters to optimize or dicts defining
            parameter groups
        lr (float): learning rate

black0017

"""
A K-NN implemented with pytorch data loading for evaluation of
self-supervised pretraining

Authors: Nikolas Adaloglou, Nima Rafiee

MIT Licence
Copyright © 2021 Nikolas Adaloglou, Nima Rafiee

Permission is hereby granted, free of charge, to any person