Minor fixes + changed the default number of epochs to 100.

[pysvrt.git] / svrt.h

diff --git a/svrt.h b/svrt.h
index 4335df3..77b8b46 100644 (file)
--- a/svrt.h
+++ b/svrt.h
@@ -23,4 +23,4 @@
  *
  */
 
-THByteTensor *generate_vignettes(long n_problem, long nb_images);
+THByteTensor *generate_vignettes(long n_problem, THLongTensor *labels);