Update app.py
Browse files
app.py
CHANGED
@@ -95,10 +95,10 @@ def load_model():
|
|
95 |
"--workers", type=int, default=16, metavar="N", help="dataloader threads"
|
96 |
)
|
97 |
parser.add_argument(
|
98 |
-
"--base-size", type=int, default=
|
99 |
)
|
100 |
parser.add_argument(
|
101 |
-
"--crop-size", type=int, default=
|
102 |
)
|
103 |
parser.add_argument(
|
104 |
"--train-split",
|
|
|
95 |
"--workers", type=int, default=16, metavar="N", help="dataloader threads"
|
96 |
)
|
97 |
parser.add_argument(
|
98 |
+
"--base-size", type=int, default=256, help="base image size"
|
99 |
)
|
100 |
parser.add_argument(
|
101 |
+
"--crop-size", type=int, default=128, help="crop image size"
|
102 |
)
|
103 |
parser.add_argument(
|
104 |
"--train-split",
|