wassname/denoising-diffusion-pytorch
1
0
Fork 0
mirror of https://github.com/wassname/denoising-diffusion-pytorch.git synced 2026-09-11 12:11:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

move accelerator backward outside of autocast context, also calculate total loss correctly across gradient accumulated steps

Browse Source
This commit is contained in:
Phil Wang
2022-07-11 21:02:06 -07:00
parent 1345a8a41d
commit 2b742dd2cc
3 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
setup.py
+1 -1
View File
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
setup(
name = 'denoising-diffusion-pytorch',
packages = find_packages(),
version = '0.25.2',
version = '0.25.3',
license='MIT',
description = 'Denoising Diffusion Probabilistic Models - Pytorch',
author = 'Phil Wang',