Spaces:
Runtime error
Runtime error
RohitGandikota
commited on
Commit
•
94e1b95
1
Parent(s):
9a7d487
fixing training
Browse files
trainscripts/textsliders/demotrain.py
CHANGED
|
@@ -426,7 +426,7 @@ def train_xl(target, positive, negative, lr, iterations, config_file, rank, devi
|
|
| 426 |
else:
|
| 427 |
attributes = []
|
| 428 |
config.network.alpha = 1.0
|
| 429 |
-
config.network.rank = rank
|
| 430 |
|
| 431 |
config.save.path += f'/{config.save.name}'
|
| 432 |
|
|
|
|
| 426 |
else:
|
| 427 |
attributes = []
|
| 428 |
config.network.alpha = 1.0
|
| 429 |
+
config.network.rank = int(rank)
|
| 430 |
|
| 431 |
config.save.path += f'/{config.save.name}'
|
| 432 |
|