Open ryanccarelli opened 2 years ago
We need to support flexible evals, like linear, knn, or other task. Evals will be specified in their own configuration files.
Different evals require different model definition (ex. removing the head in linear evals)
Different evals require different data preprocessing, as in https://github.com/deepmind/deepmind-research/blob/2c7c401024c42c4fb1aa20a8b0471d2e6b480906/byol/utils/dataset.py#L56
We need to support flexible evals, like linear, knn, or other task. Evals will be specified in their own configuration files.
Different evals require different model definition (ex. removing the head in linear evals)
Different evals require different data preprocessing, as in https://github.com/deepmind/deepmind-research/blob/2c7c401024c42c4fb1aa20a8b0471d2e6b480906/byol/utils/dataset.py#L56