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

[pysvrt.git] / README.md

diff --git a/README.md b/README.md
index 26b40f2..adae280 100644 (file)
--- a/README.md
+++ b/README.md
@@ -6,7 +6,7 @@ framework.
 The main function is
 
 ```
 The main function is
 
 ```

-torch.ByteTensor generate_vignettes(int problem_number, torch.LongTensor labels)
+torch.ByteTensor svrt.generate_vignettes(int problem_number, torch.LongTensor labels)

 ```
 
 where
 ```
 
 where


@@ -32,5 +32,5 @@ make -j -k
 should generate an image example.png in the current directory.
 
 Note that the image generation does not take advantage of GPUs or
 should generate an image example.png in the current directory.
 
 Note that the image generation does not take advantage of GPUs or

-multi-core, and can be as fast as 3,000 vignettes per second and as
+multi-core, and can be as fast as 10,000 vignettes per second and as

 slow as 40 on a 4GHz i7-6700K.
 slow as 40 on a 4GHz i7-6700K.