def parse_activation(activation_str):
if activation_str == 'relu':
return F.relu
elif activation_str == 'elu':
return F.elu
elif activation_str == 'lrelu':
return F.leaky_relu
else:
raise RuntimeError(
'Not supported activation: {}'.format(activation_str))
评论列表
文章目录