ad8ac8ff2f
Merge branch 'pytorch_integration' into pytorch_scratch
2019-10-06 17:07:54 -04:00
5b93f3c778
Merge branch 'pytorch_integration' of github.com:chnsh/DCRNN into pytorch_integration
2019-10-06 17:02:35 -04:00
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
d46b605a65
ensured row major ordering
2019-10-06 15:53:14 -04:00
6331173f44
logging and refactor
2019-10-06 15:22:57 -04:00
ec5d9555a5
logging and refactor
2019-10-06 15:15:11 -04:00
55a087ac9f
logging and refactor
2019-10-06 14:40:53 -04:00
02c4681ad9
logging and refactor
2019-10-06 14:38:44 -04:00
b6a2b3fe8e
logging and refactor
2019-10-06 14:34:58 -04:00
036e552bf6
logging and refactor
2019-10-06 14:31:46 -04:00
31acadedce
logging and refactor
2019-10-06 14:29:28 -04:00
e563e1bf37
moving tensors to GPU [v3]
2019-10-06 14:13:02 -04:00
017ec70783
moving tensors to GPU [v2]
2019-10-06 14:10:20 -04:00
ba304e9f04
moving tensors to GPU
2019-10-06 14:00:54 -04:00
9454fd91a2
Ensured sparse mm for readability, logging sparsely as well
2019-10-06 13:44:55 -04:00
2e1836df40
Rough implementation complete - could forward pass it through the network
2019-10-06 13:24:37 -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
Chintan Shah
Chintan Shah
Yaguang