From 2760d7e70c1a93cd122f1857cc6f6393a6b549a8 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Wed, 14 Jun 2017 23:01:41 +0200 Subject: [PATCH] Added the header. --- cnn-svrt.py | 24 ++++++++++++++++++++++-- 1 file changed, 22 insertions(+), 2 deletions(-) diff --git a/cnn-svrt.py b/cnn-svrt.py index d5685f4..f731c2b 100755 --- a/cnn-svrt.py +++ b/cnn-svrt.py @@ -1,5 +1,26 @@ #!/usr/bin/env python-for-pytorch +# svrt is the ``Synthetic Visual Reasoning Test'', an image +# generator for evaluating classification performance of machine +# learning systems, humans and primates. +# +# Copyright (c) 2017 Idiap Research Institute, http://www.idiap.ch/ +# Written by Francois Fleuret +# +# This file is part of svrt. +# +# svrt is free software: you can redistribute it and/or modify it +# under the terms of the GNU General Public License version 3 as +# published by the Free Software Foundation. +# +# svrt is distributed in the hope that it will be useful, but +# WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +# General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with selector. If not, see . + import time import torch @@ -11,10 +32,9 @@ from torch import nn from torch.nn import functional as fn from torchvision import datasets, transforms, utils -from _ext import svrt +import svrt ###################################################################### -# The data def generate_set(p, n): target = torch.LongTensor(n).bernoulli_(0.5) -- 2.39.5