0.5.4

use deterministic optimizer params (#116 )
2026-02-14 05:44:19 +01:00 · 2022-05-25 09:32:17 -07:00 · 2022-05-25 09:31:43 -07:00
2 changed files with 1 additions and 3 deletions
--- a/dalle2_pytorch/optimizer.py
+++ b/dalle2_pytorch/optimizer.py
@@ -21,8 +21,6 @@ def get_optimizer(
    if wd == 0:
        return Adam(params, lr = lr, betas = betas, eps = eps)

-    params = set(params)
-
    if group_wd_params:
        wd_params, no_wd_params = separate_weight_decayable_params(params)

--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@ setup(
      'dream = dalle2_pytorch.cli:dream'
    ],
  },
-  version = '0.5.3',
+  version = '0.5.4',
  license='MIT',
  description = 'DALL-E 2',
  author = 'Phil Wang',
Author	SHA1	Message	Date
Phil Wang	6161b61c55	0.5.4	2022-05-25 09:32:17 -07:00
zion	1ed0f9d80b	use deterministic optimizer params (#116 )	2022-05-25 09:31:43 -07:00