Add serde::serialize to activations. (#2732)
This commit is contained in:
parent
e4c3a71f11
commit
85f0aaefe5
|
@ -1,9 +1,8 @@
|
|||
//! Activation Functions
|
||||
//!
|
||||
use candle::{Result, Tensor};
|
||||
use serde::Deserialize;
|
||||
|
||||
#[derive(Debug, Clone, Copy, PartialEq, Deserialize, Default)]
|
||||
#[derive(Debug, Clone, Copy, PartialEq, serde::Deserialize, serde::Serialize, Default)]
|
||||
#[serde(rename_all = "lowercase")]
|
||||
pub enum Activation {
|
||||
#[default]
|
||||
|
|
Loading…
Reference in New Issue