aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-19 09:44:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

use deterministic optimizer params (#116)

Browse Source
This commit is contained in:
zion
2022-05-25 09:31:43 -07:00
committed by GitHub
parent f326a95e26
commit 1ed0f9d80b
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dalle2_pytorch/optimizer.py
View File

@@ -21,8 +21,6 @@ def get_optimizer(
if wd == 0: if wd == 0:
return Adam(params, lr = lr, betas = betas, eps = eps) return Adam(params, lr = lr, betas = betas, eps = eps)
params = set(params)
if group_wd_params: if group_wd_params:
wd_params, no_wd_params = separate_weight_decayable_params(params) wd_params, no_wd_params = separate_weight_decayable_params(params)