accept unets as list for decoder

This commit is contained in:
Phil Wang
2022-05-20 20:31:26 -07:00
parent f526f14d7c
commit 80497e9839
3 changed files with 5 additions and 2 deletions

View File

@@ -59,6 +59,9 @@ def default(val, d):
return d() if isfunction(d) else d
def cast_tuple(val, length = 1):
if isinstance(val, list):
val = tuple(val)
return val if isinstance(val, tuple) else ((val,) * length)
def module_device(module):