mirror of https://github.com/tracel-ai/burn.git
Fixed the wrong order of the attributes (#930)
Signed-off-by: GitHub <noreply@github.com>
This commit is contained in:
parent
dddc138757
commit
4e8b573f13
|
@ -47,9 +47,9 @@ impl<B: Backend> Model<B> {
|
|||
conv1,
|
||||
conv2,
|
||||
conv3,
|
||||
dropout,
|
||||
fc1,
|
||||
fc2,
|
||||
dropout,
|
||||
activation: nn::GELU::new(),
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue