DALLE2-pytorch

mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-19 09:44:19 +01:00

Files

Aidan Dempster 58892135d9 Distributed Training of the Decoder (#121 )

* Converted decoder trainer to use accelerate

* Fixed issue where metric evaluation would hang on distributed mode

* Implemented functional saving
Loading still fails due to some issue with the optimizer

* Fixed issue with loading decoders

* Fixed issue with tracker config

* Fixed issue with amp
Updated logging to be more logical

* Saving checkpoint now saves position in training as well
Fixed an issue with running out of gpu space due to loading weights into the gpu twice

* Fixed ema for distributed training

* Fixed isue where get_pkg_version was reintroduced

* Changed decoder trainer to upload config as a file

Fixed issue where loading best would error

2022-06-19 09:25:54 -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

Distributed Training of the Decoder (#121 )

2022-06-19 09:25:54 -07:00

__init__.py

fix version

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

cli.py

add missing import (#56 )