├── README.md ├── crsam_train.py ├── metrics ├── __init__.py ├── hessian.py └── metrics.py ├── models ├── __init__.py ├── resnet.py └── wrn_28_10.py └── utils ├── __init__.py ├── autoaugment.py ├── crsam.py ├── cutout.py ├── dataset.py ├── gcsam_train.py ├── logger.py └── sam.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/README.md -------------------------------------------------------------------------------- /crsam_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/crsam_train.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /metrics/hessian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/metrics/hessian.py -------------------------------------------------------------------------------- /metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/metrics/metrics.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/wrn_28_10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/models/wrn_28_10.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | from utils.sam import * 2 | -------------------------------------------------------------------------------- /utils/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/utils/autoaugment.py -------------------------------------------------------------------------------- /utils/crsam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/utils/crsam.py -------------------------------------------------------------------------------- /utils/cutout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/utils/cutout.py -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/gcsam_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/utils/gcsam_train.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TrustAIoT/CR-SAM/HEAD/utils/sam.py --------------------------------------------------------------------------------