Prechádzať zdrojové kódy

Fix issues with dense addressing

Alexey Edelev 5 rokov pred
rodič
commit
5b0133b03b
2 zmenil súbory, kde vykonal 8 pridanie a 5 odobranie
  1. 5 5
      gui/dense.cpp
  2. 3 0
      gui/visualizermodel.h

+ 5 - 5
gui/dense.cpp

@@ -44,15 +44,15 @@
 Dense::Dense(const QByteArray &data)
 {
     m_data = data;
-    m_rows = *(int64_t*)(data.data() + 8);
-    m_columns = *(int64_t*)(data.data() + 16);
+    m_rows = *(int64_t *)(data.data() + 8);
+    m_columns = *(int64_t *)(data.data() + 16);
 
-    Q_ASSERT(m_rows*m_columns + 40 == data.size());
+    Q_ASSERT(m_rows * m_columns * sizeof(double) + 40 == data.size());
 }
 
 double Dense::value(int row, int column)
 {
-    char* dataPtr = m_data.data() + 40 + m_rows * row + m_columns * column;
-    return *(double*)dataPtr;
+    char *dataPtr = m_data.data() + 40 + (m_rows * row + m_columns * column) * sizeof(double);
+    return *(double *)dataPtr;
 }
 

+ 3 - 0
gui/visualizermodel.h

@@ -40,6 +40,9 @@ public:
         return m_networkConfig.sizes();
     }
 
+//    ValueIndicator *activation(int layer, int row);
+//    ValueIndicator *weight(int layer, int row, int column);
+
 signals:
     void sizesChanged();