3 Nb@sVddlmZddlZddlmZddljZddlZddZ ddZ Gdd d eZ dS) )EnumN)EasyDictc Cs$t|d}ttj|SQRXdS)Nr)openedictyamlZ safe_load)pathfr @/home/ec2-user/SageMaker/lama/saicinpainting/evaluation/utils.py load_yaml s r cst|tjr|jStj|r*|jSt|ttfrJfdd|DSt|trjfdd|j DSt dt |dS)Ncsg|]}t|qSr )move_to_device).0el)devicer r sz"move_to_device..csi|]\}}t||qSr )r )rnameval)rr r sz"move_to_device..zUnexpected type ) isinstancennModuletotorch is_tensortuplelistdictitems ValueErrortype)objrr )rr r s     r c@seZdZdZdZdS) SmallModedropZupscaleN)__name__ __module__ __qualname__ZDROPZUPSCALEr r r r r"sr") enumrrZeasydictrrtorch.nnrrr r r"r r r r s