2020年9月18日
torch scheduler
摘要： torch scheduler class LRScheduler(object): r"""Learning Rate Scheduler Parameters mode : str Modes for learning rate scheduler. Currently it supports 阅读全文
posted @ 2020-09-18 10:56 珊珊来迟0 阅读(860) 评论(0) 推荐(0)
seg loss 相关
摘要： sigmoid focal loss class SigmoidFocalLoss(nn.Module): def __init__(self, ignore_label, gamma=2.0, alpha=0.25, reduction='mean'): super(SigmoidFocalLos 阅读全文
posted @ 2020-09-18 10:48 珊珊来迟0 阅读(458) 评论(0) 推荐(0)