Jakub Kwiatkowski commited on
Commit
bf96524
·
1 Parent(s): 76b576d

feat: Changes for new models.

Browse files
Files changed (1) hide show
  1. utils.py +7 -2
utils.py CHANGED
@@ -62,8 +62,13 @@ def run_nn(index=0):
62
  data = models.data[index:index + 1]['inputs']
63
 
64
  # model = load_model("/home/jkwiatkowski/all/best/rav/full_trans/6e8e6bad403e4171ad10daa1a518ba09")
65
- data = {
66
- 'inputs': data,
 
 
 
 
 
67
  'index': np.zeros(shape=(1, 1), dtype="uint8"),
68
  'target': np.zeros(shape=(1, 16, 113), dtype="int8"),
69
  }
 
62
  data = models.data[index:index + 1]['inputs']
63
 
64
  # model = load_model("/home/jkwiatkowski/all/best/rav/full_trans/6e8e6bad403e4171ad10daa1a518ba09")
65
+ # data = {
66
+ # 'inputs': data,
67
+ # 'index': np.zeros(shape=(1, 1), dtype="uint8"),
68
+ # 'target': np.zeros(shape=(1, 16, 113), dtype="int8"),
69
+ # }
70
+ d = {
71
+ 'inputs': np.asarray(data, dtype="uint8"),
72
  'index': np.zeros(shape=(1, 1), dtype="uint8"),
73
  'target': np.zeros(shape=(1, 16, 113), dtype="int8"),
74
  }