diff --git a/source/poet.cpp b/source/poet.cpp index de83be3..170bcdc 100644 --- a/source/poet.cpp +++ b/source/poet.cpp @@ -94,8 +94,9 @@ int main(int argc, char *argv[]) stack->train(t_vc, &listener); stack->saveWeights("."); +#if DO_SAVE_PROJECT_AFTER_TRAINING stack->save("."); - +#endif printf("Curr\n"); for (int i=0; i < t_vc.n_rows; i++) {