(parent_parser)
| 254 | |
| 255 | @staticmethod |
| 256 | def add_model_specific_args(parent_parser): |
| 257 | parser = ArgumentParser(parents=[parent_parser], add_help=False) |
| 258 | parser.add_argument("--base_encoder", type=str, default="resnet18") |
| 259 | parser.add_argument("--emb_dim", type=int, default=128) |
| 260 | parser.add_argument("--num_workers", type=int, default=8) |
| 261 | parser.add_argument("--num_negatives", type=int, default=65536) |
| 262 | parser.add_argument("--encoder_momentum", type=float, default=0.999) |
| 263 | parser.add_argument("--softmax_temperature", type=float, default=0.07) |
| 264 | parser.add_argument("--learning_rate", type=float, default=0.03) |
| 265 | parser.add_argument("--momentum", type=float, default=0.9) |
| 266 | parser.add_argument("--weight_decay", type=float, default=1e-4) |
| 267 | parser.add_argument("--data_dir", type=str, default="./") |
| 268 | parser.add_argument( |
| 269 | "--dataset", |
| 270 | type=str, |
| 271 | default="cifar10", |
| 272 | choices=["cifar10", "imagenet2012", "stl10"], |
| 273 | ) |
| 274 | parser.add_argument("--batch_size", type=int, default=256) |
| 275 | parser.add_argument("--use_mlp", action="store_true") |
| 276 | parser.add_argument( |
| 277 | "--meta_dir", default=".", type=str, help="path to meta.bin for imagenet" |
| 278 | ) |
| 279 | return parser |
| 280 | |
| 281 | @staticmethod |
| 282 | def _use_ddp_or_ddp2(trainer: Trainer) -> bool: |
nothing calls this directly
no outgoing calls
no test coverage detected