From 77153ad6f6acb94a5132e9930722500cd93a6960 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Mon, 19 Jun 2017 14:36:10 +0200 Subject: [PATCH] Update. --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 ## -- 2.20.1