[add] a linear evaluation for CLEAR benchmark
@@ -60,6 +60,14 @@ HPARAM = {
"momentum": 0.9,
}
+
+def make_scheduler(optimizer, step_size, gamma=0.1):
+ scheduler = torch.optim.lr_scheduler.StepLR(
+ optimizer, step_size=step_size, gamma=gamma
+ )
+ return scheduler
+
+
model = torchvision.models.resnet18(pretrained=False)
normalize = torchvision.transforms.Normalize(
