Add Efficientnet (#572)

* EfficientNet.

* Complete the efficientnet implementation.

* Improve group handling.

* Get the efficientnet to work.
This commit is contained in:
Laurent Mazare
2023-08-23 18:02:58 +01:00
committed by GitHub
parent eedd85ffa7
commit 431051cc32
4 changed files with 448 additions and 11 deletions

View File

@ -129,7 +129,7 @@ impl<'a> VarBuilder<'a> {
})
}
pub fn push_prefix(&self, s: &str) -> Self {
pub fn push_prefix<S: ToString>(&self, s: S) -> Self {
let mut path = self.path.clone();
path.push(s.to_string());
Self {
@ -139,7 +139,7 @@ impl<'a> VarBuilder<'a> {
}
/// Short alias for `push_prefix`.
pub fn pp(&self, s: &str) -> Self {
pub fn pp<S: ToString>(&self, s: S) -> Self {
self.push_prefix(s)
}