#6 WIP: 更新 'main.py'

Closed
zhounuoyan wants to merge 1 commits from zhounuoyan-patch-2 into master
  1. +1
    -1
      main.py

+ 1
- 1
main.py View File

@@ -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')


Loading…
Cancel
Save