gyrojeff commited on
Commit
00a4b21
·
1 Parent(s): ff82fe6

style: format

Browse files
Files changed (1) hide show
  1. train.py +2 -2
train.py CHANGED
@@ -9,10 +9,10 @@ from detector.model import FontDetector, ResNet18Regressor
9
  from utils import get_current_tag
10
 
11
 
12
- torch.set_float32_matmul_precision('high')
13
 
14
  parser = argparse.ArgumentParser()
15
- parser.add_argument('-d', '--devices', nargs='*', type=int, default=[0])
16
 
17
  args = parser.parse_args()
18
 
 
9
  from utils import get_current_tag
10
 
11
 
12
+ torch.set_float32_matmul_precision("high")
13
 
14
  parser = argparse.ArgumentParser()
15
+ parser.add_argument("-d", "--devices", nargs="*", type=int, default=[0])
16
 
17
  args = parser.parse_args()
18