Skip to content

Commit 43a563d

Browse files
committed
re-generate sample emoji
1 parent ee96073 commit 43a563d

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

artwork/growing_emoji.gif

-845 KB
Loading

tasks/growing_emoji/eval_growing_emoji.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,20 +29,23 @@ def eval_growing_emoji(gpu: bool, gpu_index: int):
2929
device,
3030
num_image_channels=4,
3131
num_hidden_channels=12,
32-
use_alive_mask=False,
32+
use_alive_mask=True,
3333
).to(device)
3434

3535
nca.load_state_dict(
3636
torch.load(WEIGHTS_PATH / "ncalab_growing_emoji.pth", weights_only=True)
3737
)
3838
nca.eval()
3939

40-
images = nca.grow(32, 32, steps=110, save_steps=True)
40+
images = nca.grow(64, 64, steps=100, save_steps=True)
4141

4242
fig, ax = plt.subplots()
4343
fig.set_size_inches(2, 2)
4444
im = ax.imshow(images[0], animated=True)
4545
ax.set_axis_off()
46+
plt.subplots_adjust(top=1, bottom=0, right=1, left=0, hspace=0, wspace=0)
47+
plt.margins(0, 0)
48+
plt.tight_layout()
4649

4750
def update(i):
4851
im.set_array(images[i])

0 commit comments

Comments
 (0)