Skip to content

Commit 1474135

Browse files
fixed global_step wrong calculation and added tuning HP notebook
1 parent 5bdb1da commit 1474135

File tree

2 files changed

+654
-4
lines changed

2 files changed

+654
-4
lines changed

finetuning_model.ipynb

+4-4
Original file line numberDiff line numberDiff line change
@@ -306,10 +306,10 @@
306306
" if writer:\n",
307307
" writer.add_scalar(\"train_loss\", \n",
308308
" running_losses/log_every_n_batches,\n",
309-
" global_step=epoch*len(train_loader)*batch_idx)\n",
309+
" global_step=epoch*len(train_loader)+batch_idx)\n",
310310
" writer.add_scalar(\"train_accuracy\", \n",
311311
" running_corrects/total,\n",
312-
" global_step=epoch*len(train_loader)*batch_idx)\n",
312+
" global_step=epoch*len(train_loader)+batch_idx)\n",
313313
" \n",
314314
" # reset to zero values for next logging\n",
315315
" running_losses = 0.0\n",
@@ -380,10 +380,10 @@
380380
" if writer:\n",
381381
" writer.add_scalar(\"val_loss\", \n",
382382
" running_losses/log_every_n_batches,\n",
383-
" global_step=epoch*len(val_loader)*batch_idx)\n",
383+
" global_step=epoch*len(val_loader)+batch_idx)\n",
384384
" writer.add_scalar(\"val_accuracy\", \n",
385385
" running_corrects/total,\n",
386-
" global_step=epoch*len(val_loader)*batch_idx)\n",
386+
" global_step=epoch*len(val_loader)+batch_idx)\n",
387387
"\n",
388388
" # reset to zero values for next logging\n",
389389
" running_losses = 0.0\n",

0 commit comments

Comments
 (0)