Skip to content

Commit 8dfcef6

Browse files
authored
Update train.py
1 parent 29e53d3 commit 8dfcef6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -82,8 +82,8 @@
8282
parser.add_argument('--triplet', action='store_true', help='use triplet loss' )
8383
parser.add_argument('--lifted', action='store_true', help='use lifted loss' )
8484
parser.add_argument('--sphere', action='store_true', help='use sphere loss' )
85-
parser.add_argument('--adv', default=0.0, type=float, help='use adversarial loss as 1.0. Follow the paper `U-turn: Crafting Adversarial Queries with Opposite-direction Features, IJCV2022`' )
86-
parser.add_argument('--aiter', default=10, type=float, help='use adversarial loss every x iter' )
85+
parser.add_argument('--adv', default=0.0, type=float, help='add the adversarial training. Follow the paper `U-turn: Crafting Adversarial Queries with Opposite-direction Features, IJCV2022`' )
86+
parser.add_argument('--aiter', default=10, type=float, help='enable adversarial loss every x iter' )
8787

8888
opt = parser.parse_args()
8989

0 commit comments

Comments
 (0)