f_end = torch.zeros(nb, height, width, dtype=torch.int64)
n = torch.arange(f_start.size(0))
- for n in range(nb):
+ for n in tqdm.tqdm(range(nb), dynamic_ncols=True, desc="world generation"):
nb_fish = torch.randint(max_nb_obj, (1,)).item() + 1
for c in torch.randperm(colors.size(0) - 2)[:nb_fish].sort().values:
i, j = (