akhaliq HF staff commited on
Commit
8185594
1 Parent(s): 115091d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=520, help="base image size"
99
  )
100
  parser.add_argument(
101
- "--crop-size", type=int, default=480, help="crop image size"
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",