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