@ -21,6 +21,8 @@ class FT:
self.model_name
)
self.model.to(self.device)
# set up optimizer
self.optimizer = torch.optim.AdamW(self.model.parameters(), lr=1e-5)
self.scheduler = ReduceLROnPlateau(