Kashif Rasul
86c89c4f58
disable wandb by default
2021-02-15 11:53:10 +01:00
Kashif Rasul
908945b422
Time grad ( #28 )
...
* initial uncond image gaussian diff
TODO make it work for multivariate vector
add conditioning
* remove tqdm
* initial unet
TODO convert to 1d conv
* initial time grad estimator
* initial training
* initial sampling
* added huber loss
* use SinusoidalPosEmb from wavegrad
* use time diff network
* fix reshaping
* fix missing property
* clip false
* updated api
* added padding
* added circular padding
* use linear schedule
* added more schedules
* added back cosine schedule
* Delete Solar-time-grad.ipynb
* updated estimator API
* not tuple
* renamed to EpsilonTheta
* removed
* added example notebook
* removed some output
* fix requirements
* formatting
* added more options to time-grad
* added article
2021-02-11 10:09:25 +01:00
Dr. Kashif Rasul
e656337f17
trying adam
2021-01-04 11:38:34 +01:00
Dr. Kashif Rasul
dee1478463
stick with one cycle for now
2021-01-03 21:51:27 +01:00
Dr. Kashif Rasul
4a90f84c16
try one cycle lr
2021-01-03 21:36:47 +01:00
Dr. Kashif Rasul
a025fa77d7
update schedular after each epoch
2021-01-03 21:21:44 +01:00
Dr. Kashif Rasul
827b496240
added schedular and gradient clipping
2021-01-03 21:11:41 +01:00
Dr. Kashif Rasul
1ac4bf70d9
added wandb
2020-12-23 16:15:50 +01:00
Dr. Kashif Rasul
b9679a8f7e
fixed trainer api
2020-12-23 12:20:26 +01:00
Dr. Kashif Rasul
2726bc94ec
added back TransformedIterableDataset
2020-12-18 13:02:26 +01:00
Dr. Kashif Rasul
b072ab227b
initial gluonts dependency
2020-12-17 17:04:56 +01:00
Dr. Kashif Rasul
cb012405bf
optimized the imports
2020-03-30 13:01:08 +02:00
Dr. Kashif Rasul
bab7adbc7e
Revert "use tqdm.auto"
...
This reverts commit ffb839dcd7 .
2020-01-31 17:22:49 +01:00
Dr. Kashif Rasul
ffb839dcd7
use tqdm.auto
2020-01-31 17:09:06 +01:00
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