Hugging Face's logo

Spaces:
Prathm
/
FashionGen
Runtime error

App Files Community
Prathm commited on
Commit
1f27f3e
·
1 Parent(s): f348140

enabled cudnn.benchmark

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ torch.set_num_threads(8)
18
 
19
  # Speed up computation
20
  torch.autograd.set_grad_enabled(True)
21
- #torch.backends.cudnn.benchmark = True
22
 
23
  # Specify model to use
24
  config = Config(
 
18
 
19
  # Speed up computation
20
  torch.autograd.set_grad_enabled(True)
21
+ torch.backends.cudnn.benchmark = True
22
 
23
  # Specify model to use
24
  config = Config(