Skip to content

Commit

Permalink
Merge pull request #31 from bokutotu/feat/rnn
Browse files Browse the repository at this point in the history
Feat/rnn
  • Loading branch information
bokutotu authored Nov 1, 2024
2 parents f48f88f + a11aeba commit 0615ab2
Show file tree
Hide file tree
Showing 172 changed files with 25,476 additions and 1,512 deletions.
850 changes: 850 additions & 0 deletions test_data_json/gru_small.json

Large diffs are not rendered by default.

12,602 changes: 12,602 additions & 0 deletions test_data_json/lstm_bidirectional.json

Large diffs are not rendered by default.

1,762 changes: 1,762 additions & 0 deletions test_data_json/lstm_fwd_bkwd_medium.json

Large diffs are not rendered by default.

268 changes: 268 additions & 0 deletions test_data_json/lstm_fwd_bkwd_small.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,268 @@
{
"rnn.weight_ih_l0": {
"shape": [
8,
2
],
"stride": [
2,
1
],
"data": [
-0.005293981172144413,
0.37932288646698,
-0.5819807648658752,
-0.5203874707221985,
-0.27234524488449097,
0.1896158903837204,
-0.014010033570230007,
0.5606575012207031,
-0.06275151669979095,
0.1871093362569809,
-0.2136969119310379,
-0.13899271190166473,
-0.6755334138870239,
-0.4683041572570801,
-0.2914857566356659,
0.026193760335445404
],
"data_type": "f32",
"ptr_offset": 0
},
"rnn.weight_hh_l0": {
"shape": [
8,
2
],
"stride": [
2,
1
],
"data": [
0.2795442044734955,
0.42428016662597656,
-0.47937673330307007,
-0.3079187273979187,
0.2568332850933075,
0.5871729254722595,
-0.14552269876003265,
0.5291363000869751,
-0.11397384107112885,
0.07482186704874039,
0.6402683854103088,
-0.6559619903564453,
-0.4451505243778229,
-0.17901486158370972,
-0.2756301760673523,
0.6109408140182495
],
"data_type": "f32",
"ptr_offset": 0
},
"rnn.bias_ih_l0": {
"shape": [
8
],
"stride": [
1
],
"data": [
-0.45833221077919006,
-0.32550448179244995,
-0.49401339888572693,
-0.6622486114501953,
-0.41276684403419495,
0.6078276038169861,
0.31552404165267944,
0.34271523356437683
],
"data_type": "f32",
"ptr_offset": 0
},
"rnn.bias_hh_l0": {
"shape": [
8
],
"stride": [
1
],
"data": [
0.0371878482401371,
-0.3625219762325287,
0.11963163316249847,
-0.6602218747138977,
-0.5109314918518066,
-0.36453473567962646,
0.44614049792289734,
0.4145916998386383
],
"data_type": "f32",
"ptr_offset": 0
},
"input": {
"shape": [
3,
1,
2
],
"stride": [
2,
2,
1
],
"data": [
0.19186942279338837,
1.2637947797775269,
-1.2904350757598877,
-0.7911027073860168,
-0.02087947353720665,
-0.7184800505638123
],
"data_type": "f32",
"ptr_offset": 0
},
"output": {
"shape": [
3,
1,
2
],
"stride": [
2,
2,
1
],
"data": [
-0.1535118669271469,
0.0033985055051743984,
-0.32110878825187683,
0.22685329616069794,
-0.2945723235607147,
0.0365699902176857
],
"data_type": "f32",
"ptr_offset": 0
},
"input_grad": {
"shape": [
3,
1,
2
],
"stride": [
2,
2,
1
],
"data": [
0.010557422414422035,
0.026116354390978813,
-0.07728055119514465,
-0.11851117014884949,
-0.015096936374902725,
-0.0308743417263031
],
"data_type": "f32",
"ptr_offset": 0
},
"rnn.weight_ih_l0_grad": {
"shape": [
8,
2
],
"stride": [
2,
1
],
"data": [
0.21715931594371796,
0.03572092950344086,
-0.09307316690683365,
-0.05358332023024559,
0.11629969626665115,
0.11310197412967682,
-0.0012049516662955284,
-0.023297177627682686,
-0.1474316269159317,
0.22660601139068604,
-0.38990604877471924,
-0.34543612599372864,
0.04021185263991356,
-0.02074984833598137,
-0.05248009040951729,
-0.038393534719944
],
"data_type": "f32",
"ptr_offset": 0
},
"rnn.weight_hh_l0_grad": {
"shape": [
8,
2
],
"stride": [
2,
1
],
"data": [
0.06499463319778442,
-0.02602470852434635,
-0.010737811215221882,
-2.880929969251156e-05,
0.03271043300628662,
-0.013741767033934593,
-0.01032883208245039,
0.007253238931298256,
-0.05449433624744415,
0.021447159349918365,
-0.15156187117099762,
0.07422848790884018,
0.028118863701820374,
-0.015438945032656193,
-0.009459368884563446,
0.0024156884755939245
],
"data_type": "f32",
"ptr_offset": 0
},
"rnn.bias_ih_l0_grad": {
"shape": [
8
],
"stride": [
1
],
"data": [
-0.4551074504852295,
0.07355727255344391,
-0.14840565621852875,
0.03238354250788689,
0.5876979231834412,
0.7409195899963379,
-0.19068323075771332,
0.05141274631023407
],
"data_type": "f32",
"ptr_offset": 0
},
"rnn.bias_hh_l0_grad": {
"shape": [
8
],
"stride": [
1
],
"data": [
-0.4551074504852295,
0.07355727255344391,
-0.14840565621852875,
0.03238354250788689,
0.5876979231834412,
0.7409195899963379,
-0.19068323075771332,
0.05141274631023407
],
"data_type": "f32",
"ptr_offset": 0
}
}
Loading

0 comments on commit 0615ab2

Please sign in to comment.