From a9b5a911c26a6577f03ca2d3d1cc388ccfaa25c3 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Wed, 18 Sep 2024 15:26:39 +0200 Subject: [PATCH] Update. --- main.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/main.py b/main.py index edc366a..6b137bf 100755 --- a/main.py +++ b/main.py @@ -658,7 +658,7 @@ for i in range(args.nb_models): nb_heads=args.nb_heads, nb_blocks=args.nb_blocks, dropout=args.dropout, - ).to(main_device) + ) # model = torch.compile(model) @@ -666,9 +666,6 @@ for i in range(args.nb_models): model.test_accuracy = 0.0 model.optimizer = torch.optim.Adam(model.parameters(), lr=args.learning_rate) - model.to(main_device).train() - optimizer_to(model.optimizer, main_device) - models.append(model) ###################################################################### -- 2.39.5