def __dropout(self,net):
net_shape = net.get_shape().as_list()
noise_shape = [net_shape[0],1,1,net_shape[-1]]
return slim.dropout(net, noise_shape=noise_shape)
微信
公众号
扫码关注公众号