github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-03 10:06:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix format (#362)

Browse Source
This commit is contained in:
Kai Wang (Victor Kai)
2022-03-10 11:33:21 +08:00
committed by Frank Lee
parent a77d73f22b
commit 53bb3bcc0a
2 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
colossalai/nn/optimizer/cpu_adam.py
View File

@@ -1,10 +1,4 @@
# modified from https://github.com/microsoft/DeepSpeed/blob/master/deepspeed/ops/adam/cpu_adam.py
import math
import torch
import time
from pathlib import Path
import colossalai
class CPUAdam(torch.optim.Optimizer):