date: 2024-08-24
title: Dropout
status: DONE
author:
- AllenYGY
tags:
- NOTE
- DeepLearning
publish: True
Dropout
一个好的模型需要对输入数据的扰动鲁棒
Tikhonov
正则import torch
def dropout_layer(X,dropout):
assert 0<= dropout <=1 #概率值在0-1之间
if dropout == 1:
return torch.zeros_like(X)
if dropout == 0:
return X
mask = (torch.randn(X.shape) > dropout).float
return mask * X / (1.0-dropout)