9fb999c3bb
squash! Added dcrnn_cell
2019-10-06 17:01:49 -04:00
d1964672c2
Added dcrnn_cell
...
Rough implementation complete - could forward pass it through the network
Ensured sparse mm for readability, logging sparsely as well
moving tensors to GPU
moving tensors to GPU [v2]
moving tensors to GPU [v3]
logging and refactor
logging and refactor
logging and refactor
logging and refactor
logging and refactor
logging and refactor
logging and refactor
ensured row major ordering
fixed log message
2019-10-06 17:00:23 -04:00
b65df994e4
Added dcrnn_cell
2019-10-06 11:55:02 -04:00
e80c47390d
Merge branch 'pytorch_implementation' into pytorch_scratch
2019-10-06 11:49:49 -04:00
5a790d5586
cuda no grad
2019-10-04 23:30:10 -04:00
593e3db1bf
Using model.cuda() if cuda is available
2019-10-04 22:45:08 -04:00
8d3b1d0d66
Implemented lr annealing schedule
2019-10-04 21:18:05 -04:00
ba880b8230
Implementing load and save models and early stopping
2019-10-04 17:25:03 -04:00
d9f41172dc
Implemented eval and function
2019-10-04 17:07:38 -04:00
20c6aa5862
Fixed bugs with refactoring
2019-10-04 16:05:52 -04:00
2b8d5e6b31
Refactored code and moved everything into a DCRNN forward pass
2019-10-04 13:02:50 -04:00
f41dc442b0
Implemented gradient clipping and returning output from training one batch
2019-10-03 19:35:54 -04:00
9834b12d5a
Cleaned up code, fixed bugs in implementation, seems like it starts training with GRU
2019-10-02 22:20:43 -04:00
f96a8c0d59
Dirty commit - setup model but [GRUCell] not working, tried ParameterList, did not work
2019-10-02 18:09:33 -04:00
c876cbfba3
Setup training loop and logging
2019-10-02 17:34:07 -04:00
86c4c5704d
Implemented single batch forward pass
2019-10-01 22:47:59 -04:00
69288460b1
Simplified encoder decoder model and moved curriculum learning outside
2019-10-01 10:07:34 -04:00
a1c9af2bad
Setup curriculum learning framework
2019-09-30 21:58:55 -04:00
bdce241a8f
Implemented abstract method and changed scheme to do all layers first for each timestep
2019-09-30 20:32:31 -04:00
0769a3b2e2
Implemented seq2seq without DCGRU and curriculum learning
2019-09-30 19:14:24 -04:00
7349f2ed67
Figured out decoder shapes
2019-09-30 16:04:03 -04:00
6386ac7eb4
Implemented encoder using GRUCell instead so that it's easier to swap that with DCGRUCell
2019-09-29 17:40:52 -04:00
66fb202d21
Implemented Encoder with GRU - should swap GRU with DCGRU
2019-09-29 12:51:49 -04:00
7a5e3c0216
model partially implemented - does not work yet
2019-09-29 11:13:08 -04:00
adbfa19146
Implemented DCGRUCell in pytorch (untested)
2019-09-08 19:28:20 -04:00
00c70b3a27
Implemented fc layer and changed docker image to use pytorch
2019-09-08 18:47:19 -04:00
7ba7fa320d
Using pytorch image
2019-09-07 17:53:46 -04:00
69d6c0e053
Add Dockerfile
2019-09-07 17:18:31 -04:00
8c20ca1a9c
Adds baseline methods for evaluation.
2019-06-18 13:00:24 -07:00
37f0734cb4
Update README.md
2019-03-18 21:32:57 -07:00
3caf887484
Add data example.
2019-03-17 11:56:55 -07:00
4344dddc6d
Add instructions on using HDF5 with python.
2019-03-17 11:43:30 -07:00
81b4626193
Remove unused code.
2019-01-18 19:00:34 -08:00
734ecbc138
Update README.md
2019-01-12 12:04:28 -08:00
763eb7af69
Add pretrained model on PEMS-BAY.
2019-01-10 17:44:07 -08:00
ad36deb794
Update README.md
...
Add more instructions.
2019-01-08 11:33:20 -08:00
c8a676604b
Add PEMS-BAY configuration.
2019-01-08 11:21:22 -08:00
1481b24b1b
Update README.md
...
Fix typo in the Graph Construction script (thanks to Zifeng Kang).
2018-12-24 00:47:17 -08:00
ac5321fcf7
Update README.md
...
Add an additional link for data sharing.
2018-11-03 11:54:22 -07:00
a8aa1732a9
Update README.md
2018-10-12 13:30:53 -07:00
d59d44e4f0
Code refactor.
2018-10-01 17:45:46 -07:00
bbc06b6c0c
Add log level support and load_dataset method.
2018-10-01 17:45:28 -07:00
88d9fc86d1
Update instructions for graph generation.
2018-10-01 10:54:56 -07:00
e6645df191
Update README.
2018-10-01 09:59:47 -07:00
2e4b8c868f
Merge pull request #9 from liyaguang/v2
...
Upgrading to DCRNN v2.
2018-10-01 09:56:22 -07:00
9520e6cf85
Update pretrained model.
2018-10-01 09:47:23 -07:00
e0212cc178
Update README and requirements.
2018-09-30 22:15:27 -07:00
7e05414d99
Combine Val and Test model.
2018-09-30 22:14:45 -07:00
5dc36fed7c
Refactor DCRNN Model.
2018-09-30 21:53:40 -07:00
2f2d748b45
Add AMSGrad from https://github.com/taki0112/AMSGrad-Tensorflow for stablized Adam Training.
2018-09-30 21:52:20 -07:00
Chintan Shah
Chintan Shah
Yaguang