projects
/
culture.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[culture.git]
/
world.py
diff --git
a/world.py
b/world.py
index
ac201e7
..
43126d5
100755
(executable)
--- a/
world.py
+++ b/
world.py
@@
-34,7
+34,7
@@
def generate(
nb,
height,
width,
nb,
height,
width,
- max_nb_obj=
len(colors) -
2,
+ max_nb_obj=2,
nb_iterations=2,
):
f_start = torch.zeros(nb, height, width, dtype=torch.int64)
nb_iterations=2,
):
f_start = torch.zeros(nb, height, width, dtype=torch.int64)
@@
-43,7
+43,7
@@
def generate(
for n in range(nb):
nb_fish = torch.randint(max_nb_obj, (1,)).item() + 1
for n in range(nb):
nb_fish = torch.randint(max_nb_obj, (1,)).item() + 1
- for c in
range(nb_fish)
:
+ for c in
torch.randperm(colors.size(0) - 2)[:nb_fish].sort().values
:
i, j = (
torch.randint(height - 2, (1,))[0] + 1,
torch.randint(width - 2, (1,))[0] + 1,
i, j = (
torch.randint(height - 2, (1,))[0] + 1,
torch.randint(width - 2, (1,))[0] + 1,
@@
-117,7
+117,7
@@
if __name__ == "__main__":
height, width = 6, 8
start_time = time.perf_counter()
height, width = 6, 8
start_time = time.perf_counter()
- seq = generate(nb=64, height=height, width=width)
+ seq = generate(nb=64, height=height, width=width
, max_nb_obj=3
)
delay = time.perf_counter() - start_time
print(f"{seq.size(0)/delay:02f} samples/s")
delay = time.perf_counter() - start_time
print(f"{seq.size(0)/delay:02f} samples/s")