Commit
·
0982d38
1
Parent(s):
58bec3f
update notebook
Browse files- train.ipynb +0 -3
train.ipynb
CHANGED
|
@@ -90,9 +90,6 @@
|
|
| 90 |
" per_device_train_batch_size = int(os.getenv(\"PER_DEVICE_TRAIN_BATCH_SIZE\", 256))\n",
|
| 91 |
" max_saved_checkpoints = int(os.getenv(\"MAX_SAVED_CHECKPOINTS\", 8))\n",
|
| 92 |
" patience = max_saved_checkpoints - 1\n",
|
| 93 |
-
" n_eval_samples = int(os.getenv(\"N_EVAL_SAMPLES\", 5)) # Number of samples to show\n",
|
| 94 |
-
" log_predictions_to_wandb = os.getenv(\"LOG_PREDICTIONS_TO_WANDB\", \"true\").lower() == \"true\"\n",
|
| 95 |
-
" log_predictions_to_console = os.getenv(\"LOG_PREDICTIONS_TO_CONSOLE\", \"false\").lower() == \"true\"\n",
|
| 96 |
"\n",
|
| 97 |
" num_processes = torch.cuda.device_count()\n",
|
| 98 |
" \n",
|
|
|
|
| 90 |
" per_device_train_batch_size = int(os.getenv(\"PER_DEVICE_TRAIN_BATCH_SIZE\", 256))\n",
|
| 91 |
" max_saved_checkpoints = int(os.getenv(\"MAX_SAVED_CHECKPOINTS\", 8))\n",
|
| 92 |
" patience = max_saved_checkpoints - 1\n",
|
|
|
|
|
|
|
|
|
|
| 93 |
"\n",
|
| 94 |
" num_processes = torch.cuda.device_count()\n",
|
| 95 |
" \n",
|