mirror of
https://github.com/huggingface/candle.git
synced 2025-06-18 19:47:12 +00:00
Add Efficientnet (#572)
* EfficientNet. * Complete the efficientnet implementation. * Improve group handling. * Get the efficientnet to work.
This commit is contained in:
@ -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)
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user