From 119d5e84350bcab97c06a5c30227a294ebadb3c3 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Fri, 1 Jul 2022 10:01:27 +0200 Subject: [PATCH] Update. --- picoclvr.py | 1 + 1 file changed, 1 insertion(+) diff --git a/picoclvr.py b/picoclvr.py index 712da17..601bdf7 100755 --- a/picoclvr.py +++ b/picoclvr.py @@ -83,6 +83,7 @@ def generate(nb, height = 6, width = 8, nb_squares = torch.randint(max_nb_squares, (1,)) + 1 square_position = torch.randperm(height * width)[:nb_squares] + # color 0 is white and reserved for the background square_c = torch.randperm(nb_colors)[:nb_squares] + 1 square_i = square_position.div(width, rounding_mode = 'floor') square_j = square_position % width -- 2.20.1