This commit is contained in:
@@ -14,7 +14,7 @@ pub struct LIFConfig {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl LIFConfig {
|
impl LIFConfig {
|
||||||
pub fn init<B: Backend, const D: usize>(&self) -> LIF {
|
pub fn init<const D: usize>(&self) -> LIF {
|
||||||
LIF {
|
LIF {
|
||||||
beta: self.beta,
|
beta: self.beta,
|
||||||
threshold: self.threshold,
|
threshold: self.threshold,
|
||||||
@@ -24,6 +24,7 @@ impl LIFConfig {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// TODO: tensor cloning and its lifecycle is probably wrong, may cause comp graph to drop. Refer burn example to find proper tensor handling..
|
// TODO: tensor cloning and its lifecycle is probably wrong, may cause comp graph to drop. Refer burn example to find proper tensor handling..
|
||||||
|
#[allow(clippy::upper_case_acronyms)]
|
||||||
#[derive(Debug, Module, Clone)]
|
#[derive(Debug, Module, Clone)]
|
||||||
pub struct LIF {
|
pub struct LIF {
|
||||||
beta: f32,
|
beta: f32,
|
||||||
|
|||||||
Reference in New Issue
Block a user