From: Francois Fleuret Date: Mon, 19 Jun 2017 12:36:10 +0000 (+0200) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=77153ad6f6acb94a5132e9930722500cd93a6960;p=pysvrt.git Update. --- diff --git a/README.md b/README.md index b6e29c7..d0ec215 100644 --- a/README.md +++ b/README.md @@ -27,13 +27,13 @@ The svrtset.py implements the classes `VignetteSet` and `CompressedVignetteSet` with the following constructor ``` -def __init__(self, problem_number, nb_samples, batch_size, cuda = False, logger = None): +__init__(problem_number, nb_samples, batch_size, cuda = False, logger = None) ``` and the following method to return one batch ``` -def get_batch(self, b): +(torch.FloatTensor, torch.LongTensor) get_batch(b) ``` ## Low-level functions ##