# 210902 피어세션 ## 앙상블 시도 EffNet b7 - 석민: Focal_smoothing, gender - 준희: Focal_smoothing, mask - 태욱: Focal_smoothing, age EffNet b7 - 하은: Focal, Mask - 종원: Focal, Age - 태현: Focal, gender - 지호: cutmix + mixup + facecrop ## 내일 할 것 - 랩업 리포트 - 코드 리팩토링 ``` python train.py \ --epochs 5 \ --dataset MaskSplitStratifyDataset \ --label mask/age/gender \ --augmentation get_transforms \ --resize 384 256 \ --lr 0.00003 \ --lr_decay_step 50 \ --lr_gamma 0.5 \ --batch_size 16 \ --valid_batch_size 32 \ --model EfficientNet \ --model_version tf_efficientnet_b7\ --optimizer Adam \ --criterion focal/focal_smoothing\ --log_interval 100\ --name T2065/EfficientNet_b7_focal_smoothing_0902_mask ``` ``` python train.py \ --epochs 5 \ --dataset MaskSplitStratifyDataset \ --label mask/age/gender \ --augmentation get_transforms \ --resize 384 256 \ --lr 0.00003 \ --lr_decay_step 50 \ --lr_gamma 0.5 \ --batch_size 16 \ --valid_batch_size 32 \ --model resnet50 \ --model_version tf_efficientnet_b7\ --optimizer Adam \ --criterion focal/focal_smoothing\ --log_interval 100\ --name T2065/EfficientNet_b7_focal_smoothing_0902_mask ``` ``` python inference.py \ --batch_size 32 \ --resize 384 256 \ --model EfficientNet \ --model_version tf_efficientnet_b7 \ --name T2065/EfficientNet_b7_focal_smoothing_0902_mask \ --output_name EfficientNet_b7_focal_smoothing_0902_mask ``` ``` inference.py def inference(data_dir, model_dir, output_dir, args): """ """ use_cuda = torch.cuda.is_available() device = torch.device("cuda" if use_cuda else "cpu") num_classes = 2 # 18 <<- 요거 바꾸기 마스크는 3 ```