Aidan Dempster a71f693a26 Add the ability to auto restart the last run when started after a crash (#191) ...

* Added autoresume after crash functionality to the trackers

* Updated documentation

* Clarified what goes in the autorestart object

* Fixed style issues

Unraveled conditional block

Chnaged to using helper function to get step count

2022-07-08 13:35:40 -07:00

..

data

bring in the simple tokenizer released by openai, but also plan on leaving room for custom tokenizer with yttm

2022-04-12 09:23:17 -07:00

dataloaders

Add data flexibility to decoder trainer (#165)

2022-06-25 19:05:20 -07:00

__init__.py

fix version

2022-05-30 11:06:34 -07:00

cli.py

add missing import (#56)

2022-05-04 07:42:20 -07:00

dalle2_pytorch.py

allow for control over use of nearest interp method of downsampling low res conditioning, in addition to being able to turn it off

2022-07-08 11:44:43 -07:00

optimizer.py

lower beta2 for adam down to 0.99, based on https://openreview.net/forum?id=2LdBqxc1Yv

2022-06-03 10:26:28 -07:00

tokenizer.py

make youtokentome optional package, due to reported installation difficulties

2022-06-01 09:25:35 -07:00

trackers.py

Add the ability to auto restart the last run when started after a crash (#191)

2022-07-08 13:35:40 -07:00

train_configs.py

Add the ability to auto restart the last run when started after a crash (#191)

2022-07-08 13:35:40 -07:00

trainer.py

Add the ability to auto restart the last run when started after a crash (#191)

2022-07-08 13:35:40 -07:00

utils.py

Overhauled the tracker system (#172)

2022-07-01 09:39:40 -07:00

version.py

expose num_steps_taken helper method on trainer to retrieve number of training steps of each unet

2022-07-08 13:00:56 -07:00

vqgan_vae_trainer.py

EMA for vqgan vae comes from ema_pytorch now

2022-06-20 15:29:08 -07:00

vqgan_vae.py

changed str in order to avoid confusions and collisions with Python (#147)

2022-06-15 13:41:16 -07:00