johnsutor/emoji-painter 🖼️🔢 → 🖼️

device: cuda
hidden_dim: 256
log_interval: 100
log_with: tensorboard
mixed_precision: 'no'
n_dec_layers: 3
n_enc_layers: 3
n_heads: 8
num_blocks: 3
num_shapes: 501
num_strokes: 8
param_per_stroke: 4
param_shape: 4
patch_size: 64
recreation_path: !!python/object/apply:cog.types.Path
- /
- tmp
- tmpbkt1i4ou
- output.png
scale: 6.0
shapes_path: ./shapes.pth
target_path: !!python/object/apply:cog.types.Path
- /
- tmp
- tmpn7xxyyrsmona.jpg
weights_path: ./weights.pth
hydra:
run:
dir: ./
  0%|          | 0/7 [00:00<?, ?it/s]/root/.pyenv/versions/3.10.14/lib/python3.10/site-packages/torch/functional.py:507: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3549.)
return _VF.meshgrid(tensors, **kwargs)  # type: ignore[attr-defined]
 14%|█▍        | 1/7 [00:02<00:15,  2.55s/it]
 29%|██▊       | 2/7 [00:07<00:20,  4.04s/it]
 43%|████▎     | 3/7 [00:13<00:19,  4.99s/it]
 57%|█████▋    | 4/7 [00:16<00:11,  3.97s/it]
 71%|███████▏  | 5/7 [00:17<00:05,  2.92s/it]
 86%|████████▌ | 6/7 [00:18<00:02,  2.31s/it]
100%|██████████| 7/7 [00:19<00:00,  1.91s/it]
100%|██████████| 7/7 [00:19<00:00,  2.78s/it]