Dr. Kashif Rasul
|
79fba1fc26
|
log weights at the end of each epoch
|
2020-01-16 18:24:05 +01:00 |
|
Dr. Kashif Rasul
|
616ca73a68
|
comment out graph for now and close it at the end
|
2020-01-16 15:17:37 +01:00 |
|
Dr. Kashif Rasul
|
994f1202b7
|
added initial SummaryWriter
|
2020-01-16 14:34:50 +01:00 |
|
Dr. Kashif Rasul
|
891aec4812
|
pin memory false by default
|
2020-01-16 10:59:40 +01:00 |
|
Dr. Kashif Rasul
|
5eacc141ac
|
typo
|
2020-01-16 10:33:52 +01:00 |
|
Dr. Kashif Rasul
|
00d790a66d
|
use pytorch Dataloader and TransformedIterableDataset
|
2020-01-16 10:32:30 +01:00 |
|
Dr. Kashif Rasul
|
eba1f50a6a
|
add weight_decay to adam
|
2020-01-14 17:27:50 +01:00 |
|
Dr. Kashif Rasul
|
032f32faf0
|
black
|
2019-12-21 14:59:11 +01:00 |
|
Dr. Kashif Rasul
|
df81b1ee97
|
print epoch_no while training
|
2019-12-17 15:03:29 +01:00 |
|
Dr. Kashif Rasul
|
aa5d3962b9
|
no validation for now
|
2019-12-17 13:47:07 +01:00 |
|
Dr. Kashif Rasul
|
2f1ae46ce8
|
avg epoch loss
|
2019-12-17 13:41:16 +01:00 |
|
Dr. Kashif Rasul
|
ab8f9f054e
|
fixed imports
|
2019-11-01 13:17:41 +01:00 |
|
Kashif Rasul
|
5772c04ffe
|
ran black
|
2019-10-30 09:42:02 +01:00 |
|
Dr. Kashif Rasul
|
a3802d6585
|
initial trainer
|
2019-10-11 14:50:59 +02:00 |
|