From: Francois Fleuret <francois@fleuret.org>
Date: Fri, 1 Jul 2022 08:01:27 +0000 (+0200)
Subject: Update.
X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=mygpt.git;a=commitdiff_plain;h=119d5e84350bcab97c06a5c30227a294ebadb3c3

Update.
---

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