This commit is contained in:
@@ -14,7 +14,7 @@ pub struct LIFConfig {
|
||||
}
|
||||
|
||||
impl LIFConfig {
|
||||
pub fn init<B: Backend, const D: usize>(&self) -> LIF {
|
||||
pub fn init<const D: usize>(&self) -> LIF {
|
||||
LIF {
|
||||
beta: self.beta,
|
||||
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..
|
||||
#[allow(clippy::upper_case_acronyms)]
|
||||
#[derive(Debug, Module, Clone)]
|
||||
pub struct LIF {
|
||||
beta: f32,
|
||||
|
||||
Reference in New Issue
Block a user