|
306 | 306 | " if writer:\n",
|
307 | 307 | " writer.add_scalar(\"train_loss\", \n",
|
308 | 308 | " 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", |
310 | 310 | " writer.add_scalar(\"train_accuracy\", \n",
|
311 | 311 | " running_corrects/total,\n",
|
312 |
| - " global_step=epoch*len(train_loader)*batch_idx)\n", |
| 312 | + " global_step=epoch*len(train_loader)+batch_idx)\n", |
313 | 313 | " \n",
|
314 | 314 | " # reset to zero values for next logging\n",
|
315 | 315 | " running_losses = 0.0\n",
|
|
380 | 380 | " if writer:\n",
|
381 | 381 | " writer.add_scalar(\"val_loss\", \n",
|
382 | 382 | " 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", |
384 | 384 | " writer.add_scalar(\"val_accuracy\", \n",
|
385 | 385 | " running_corrects/total,\n",
|
386 |
| - " global_step=epoch*len(val_loader)*batch_idx)\n", |
| 386 | + " global_step=epoch*len(val_loader)+batch_idx)\n", |
387 | 387 | "\n",
|
388 | 388 | " # reset to zero values for next logging\n",
|
389 | 389 | " running_losses = 0.0\n",
|
|
0 commit comments