Torchélie
latest
Pytorch Utils
torchelie.nn
torchelie.nn.utils
torchelie.optim
torchelie.lr_scheduler
torchelie.utils
torchelie.data_learning
torchelie.loss
torchelie.models
VGG
Resnets
Pix2Pix
Pix2PixHD
StyleGAN2
Other GANs
Image classifiers
Attention56Bone
attention56
EfficientNet
Hourglass
Image Transformer
Classification heads
PixelCNN
torchelie.distributions
torchelie.datasets
torchelie.transforms
torchelie.recipes
Algorithms and training
Understanding recipes
Using the predefined recipes and callbacks
torchelie.callbacks
torchelie.hyper
Torchélie
Docs
»
torchelie.models
»
EfficientNet
Edit on GitHub
EfficientNet
¶
class
torchelie.models.
EfficientNet
(
in_ch
,
num_classes
,
B
=
0
)
¶
training
:
bool
¶