Spaces:
Sleeping
Sleeping
Commit
•
a70d1ae
1
Parent(s):
2f6ff4f
Update app.py
Browse files
app.py
CHANGED
@@ -67,7 +67,7 @@ config = {
|
|
67 |
'batch_size':3,
|
68 |
'epochs': 1,
|
69 |
'lr': 1e-5,
|
70 |
-
'device':
|
71 |
'scheduler': 'CosineAnnealingWarmRestarts'
|
72 |
}
|
73 |
|
|
|
67 |
'batch_size':3,
|
68 |
'epochs': 1,
|
69 |
'lr': 1e-5,
|
70 |
+
'device': 'cpu',
|
71 |
'scheduler': 'CosineAnnealingWarmRestarts'
|
72 |
}
|
73 |
|