Skip to content

GitLab

Commit 3fd3e2c8 authored by Kexin Yu's avatar Kexin Yu
Browse files

add import

parent c8bcfff8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
apex/optimizers/fused_lamb.py
View file @ 3fd3e2c8
import torch import torch
import math
from apex.multi_tensor_apply import multi_tensor_applier from apex.multi_tensor_apply import multi_tensor_applier
class FusedLAMB(torch.optim.Optimizer): class FusedLAMB(torch.optim.Optimizer):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment