layout_data.models package

Submodules

layout_data.models.focalloss module

class layout_data.models.focalloss.FocalLoss(gamma=2, alpha=0.25)[源代码]

基类:sphinx.ext.autodoc.importer._MockObject

forward(x, target)[源代码]

layout_data.models.scheduler module

@author: songanjie @contact: anjie_song@shannonai.com

@version: 1.0 @file: scheduler.py @time: 2019-07-17 16:16

class layout_data.models.scheduler.WarmupMultiStepLR(optimizer, milestones, gamma=0.1, warmup_factor=0.3333333333333333, warmup_iters=500, warmup_method='linear', last_epoch=-1)[源代码]

基类:sphinx.ext.autodoc.importer._MockObject

get_lr()[源代码]

Module contents