🦠️ Neural Cellular Automata ⚡️

Built with PyTorch Lightning

🔬  Why?

• PyTorch Lightning implementations are clean
• I wanted to learn about these for fun

🧪  Get Started

1. Clone this repository

2. Install the requirements pip3 install -r requirements.txt

3. Run the initial example in python3.9 nca/example_texture_nca.py which is a port of the simple pytorch example for Self-Organising Textures

   python3.9 nca/example_texture_nca.py --train_steps=500 run --lr=0.0005
   #                                                       /
   #  'run' separates trainer args from NCA args ---------/

   See all available arguments:

   • General Training Arguments

     python3.9 nca/example_texture_nca.py --help

     --train_steps=TRAIN_STEPS
         Type: int
         Default: 5000
     --train_cuda=TRAIN_CUDA
         Type: bool
         Default: False
     --vis_period_plt=VIS_PERIOD_PLT
         Type: int
         Default: 500
     --vis_period_vid=VIS_PERIOD_VID
         Type: int
         Default: 2500
     --vis_im_size=VIS_IM_SIZE
         Type: int
         Default: 256
     --vis_out_dir=VIS_OUT_DIR
         Type: str
         Default: 'out/2021-06-11_22-09-29'
     --plt_show=PLT_SHOW
         Type: bool
         Default: False
     

   • NCA Arguments

     python3.9 nca/example_texture_nca.py run --help
     

     --style_img_uri=STYLE_IMG_URI
         Type: str
         Default: 'https://upload.wikimedia.org/wikipedia/commons/thumb/0/04/T...
     --style_img_size=STYLE_IMG_SIZE
         Type: int
         Default: 128
     --nca_img_size=NCA_IMG_SIZE
         Type: int
         Default: 128
     --nca_learn_filters=NCA_LEARN_FILTERS
         Type: bool
         Default: False
     --nca_channels=NCA_CHANNELS
         Type: int
         Default: 12
     --nca_hidden_channels=NCA_HIDDEN_CHANNELS
         Type: int
         Default: 96
     --nca_pad_mode=NCA_PAD_MODE
         Type: str
         Default: 'circular'
     --nca_default_update_ratio=NCA_DEFAULT_UPDATE_RATIO
         Type: float
         Default: 0.5
     --iters=ITERS
         Type: typing.Tuple[int, int]
         Default: (32, 64)
     --batch_size=BATCH_SIZE
         Type: int
         Default: 4
     --lr=LR
         Type: float
         Default: 0.001
     --pool_size=POOL_SIZE
         Type: int
         Default: 1024
     --pool_reset_element_period=POOL_RESET_ELEMENT_PERIOD
         Type: int
         Default: 2
     --pool_on_cpu=POOL_ON_CPU
         Type: bool
         Default: True
     --normalize_gradient=NORMALIZE_GRADIENT
         Type: bool
         Default: True
     --consistency_loss_scale=CONSISTENCY_LOSS_SCALE
         Type: Optional[float]
         Default: None
     --scale_loss=SCALE_LOSS
         Type: Optional[float]
         Default: None