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
|
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 |