Clean unused imports

train_weak_to_strong.py

@@ -1,7 +1,5 @@
-import functools
 import json
 import os
-import time
 from typing import Dict, List, Optional, Sequence, Union
 
 import fire

vision/run_weak_strong.py

@@ -1,8 +1,6 @@
-import einops
 import fire
 import numpy as np
 import torch
-import torchvision
 import tqdm
 from data import get_imagenet
 from models import alexnet, resnet50_dino, vitb8_dino

weak_to_strong/datasets.py

@@ -3,10 +3,8 @@
 from random import Random
 from typing import Any, Callable, Optional
 
-import torch
 from datasets import Dataset as HfDataset
 from datasets import load_dataset as hf_load_dataset
-from transformers import GPT2Tokenizer
 
 
 @dataclass

weak_to_strong/logger.py

@@ -1,7 +1,6 @@
 import json
 import os
 from datetime import datetime
-from typing import List
 
 import wandb
 

weak_to_strong/loss.py

@@ -1,5 +1,3 @@
-from typing import Callable, Optional
-
 import torch