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

[pysvrt.git] / README.md

diff --git a/README.md b/README.md
index 450ccd2..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 slow as 40 vignettes per second s on a 4GHz
-i7-6700K.
+multi-core, and can be as fast as 10,000 vignettes per second and as
+slow as 40 on a 4GHz i7-6700K.