wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-08-09 12:20:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

log weights at the end of each epoch

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-16 18:24:05 +01:00
parent 616ca73a68
commit 79fba1fc26
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/trainer.py
+2 -2
View File
@@ -65,13 +65,13 @@ class Trainer:
)
n_iter = epoch_no*self.num_batches_per_epoch + batch_no
writer.add_scalar('Loss/train', loss.item(), n_iter)
for name, param in net.named_parameters():
writer.add_histogram(name, param.clone().cpu().data.numpy(), n_iter)
loss.backward()
optimizer.step()
if self.num_batches_per_epoch == batch_no:
for name, param in net.named_parameters():
writer.add_histogram(name, param.clone().cpu().data.numpy(), n_iter)
break
# mark epoch end time and log time cost of current epoch