DBN__destruct

michaelxu1107 · Feb 11, 2013 · c4dbea6 · c4dbea6
1 parent 876e9ff
commit c4dbea6
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/c/DBN.c b/c/DBN.c
@@ -74,7 +74,13 @@ void DBN__construct(DBN* this, int N, \
 }
 
 void DBN__destruct(DBN* this) {
-
+  int i;
+  for(i=0; i<this->n_layers; i++) {
+    HiddenLayer__destruct(&(this->sigmoid_layers[i]));
+    RBM__destruct(&(this->rbm_layers[i]));
+  }
+  free(this->sigmoid_layers);
+  free(this->rbm_layers);
 }
 
 void DBN_pretrain(DBN* this, int *input, double lr, int k, int epochs) {
@@ -561,13 +567,15 @@ void test_dbn(void) {
     }
     printf("\n");
   }
+
+  // destruct DBN
+  DBN__destruct(&dbn);
 
 }
 
 
 
 int main(void) {
   test_dbn();
-
   return 0;
 }