rev

cvlab-yonsei · Feb 4, 2021 · 744a633 · 744a633
1 parent 2616712
commit 744a633
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Evaluate.py b/Evaluate.py
@@ -37,7 +37,7 @@
 parser.add_argument('--h', type=int, default=256, help='height of input images')
 parser.add_argument('--w', type=int, default=256, help='width of input images')
 parser.add_argument('--c', type=int, default=3, help='channel of input images')
-parser.add_argument('--method', type=str, default='prediction', help='The target task for anoamly detection')
+parser.add_argument('--method', type=str, default='pred', help='The target task for anoamly detection')
 parser.add_argument('--t_length', type=int, default=5, help='length of the frame sequences')
 parser.add_argument('--fdim', type=int, default=512, help='channel dimension of the features')
 parser.add_argument('--mdim', type=int, default=512, help='channel dimension of the memory items')

diff --git a/Train.py b/Train.py
@@ -38,7 +38,7 @@
 parser.add_argument('--w', type=int, default=256, help='width of input images')
 parser.add_argument('--c', type=int, default=3, help='channel of input images')
 parser.add_argument('--lr', type=float, default=2e-4, help='initial learning rate')
-parser.add_argument('--method', type=str, default='prediction', help='The target task for anoamly detection')
+parser.add_argument('--method', type=str, default='pred', help='The target task for anoamly detection')
 parser.add_argument('--t_length', type=int, default=5, help='length of the frame sequences')
 parser.add_argument('--fdim', type=int, default=512, help='channel dimension of the features')
 parser.add_argument('--mdim', type=int, default=512, help='channel dimension of the memory items')