allow for division of loss prior to scaling, for gradient accumulation purposes

2026-02-23 22:14:20 +01:00 · 2022-04-30 12:56:47 -07:00
parent a2ef69af66
commit 63195cc2cb
2 changed files with 10 additions and 3 deletions
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@ setup(
      'dream = dalle2_pytorch.cli:dream'
    ],
  },
-  version = '0.0.79',
+  version = '0.0.80',
  license='MIT',
  description = 'DALL-E 2',
  author = 'Phil Wang',