From: François Fleuret Date: Wed, 18 Sep 2024 13:26:39 +0000 (+0200) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=a9b5a911c26a6577f03ca2d3d1cc388ccfaa25c3;p=culture.git Update. --- 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) ######################################################################