feat: commiting model
This commit is contained in:
@@ -44,7 +44,7 @@
|
||||
"BATCH_SIZE = 64\n",
|
||||
"EPOCHS = 15\n",
|
||||
"LR = 3e-4\n",
|
||||
"NUM_WORKERS = 40\n",
|
||||
"NUM_WORKERS = 62\n",
|
||||
"\n",
|
||||
"device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
|
||||
"print(f\"Аппаратное ускорение: {device}\")"
|
||||
|
||||
Reference in New Issue
Block a user