Spaces:
Runtime error
Runtime error
File size: 547 Bytes
8e5cc83 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# EfficientViT: Multi-Scale Linear Attention for High-Resolution Dense Prediction
# Han Cai, Junyan Li, Muyan Hu, Chuang Gan, Song Han
# International Conference on Computer Vision (ICCV), 2023
from efficientvit.apps.trainer.run_config import RunConfig
__all__ = ["ClsRunConfig"]
class ClsRunConfig(RunConfig):
label_smooth: float
mixup_config: dict # allow none to turn off mixup
bce: bool
mesa: dict
@property
def none_allowed(self):
return ["mixup_config", "mesa"] + super().none_allowed
|