Skip to content

Commit 2995488

Browse files
author
a-silvestrov
committed
classes int to long fix
1 parent da7e14b commit 2995488

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

task1_20newsgroups.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,7 @@
283283
"model = nn.Linear(UNIQUE_WORDS_N, UNIQUE_LABELS_N)\n",
284284
"\n",
285285
"scheduler = lambda optim: \\\n",
286-
" torch.optim.lr_scheduler.ReduceLROnPlateau(optim, patience=5, factor=0.5, verbose=True))\n",
286+
" torch.optim.lr_scheduler.ReduceLROnPlateau(optim, patience=5, factor=0.5, verbose=True)\n",
287287
"\n",
288288
"best_val_loss, best_model = train_eval_loop(model=model,\n",
289289
" train_dataset=train_dataset,\n",

0 commit comments

Comments
 (0)