10 lines
339 B
Python
10 lines
339 B
Python
|
import gc
|
||
|
|
import traceback
|
||
|
|
import torch
|
||
|
|
|
||
|
|
# This iterator returns the extensions in the order specified in the command-line
|
||
|
|
def iterator():
|
||
|
|
state_extensions = {}
|
||
|
|
for name in sorted(state_extensions, key=lambda x: state_extensions[x][1]):
|
||
|
|
if state_extensions[name][0]:
|
||
|
|
yield getattr(extensions, name).script, name
|