diff --git a/main.py b/main.py index bcd7053..cd4c77d 100644 --- a/main.py +++ b/main.py @@ -64,7 +64,7 @@ parser.add_argument('--noise-info', default=None, type=str, help='directory of pre-configured noise pattern.') parser.add_argument('--use-refined-label', action='store_true', help='whether or not use refined label by self-adaptive training') parser.add_argument('--turn-off-aug', action='store_true', help='whether or not use data augmentation') -# loss function + parser.add_argument('--loss', default='ce', help='loss function') parser.add_argument('--sat-alpha', default=0.9, type=float, help='momentum term of self-adaptive training')