Module: tfm.optimization.lars_optimizer

Stay organized with collections Save and categorize content based on your preferences.

Layer-wise adaptive rate scaling optimizer.

Classes

class LARS: Layer-wise Adaptive Rate Scaling for large batch training.