kevinwang676's picture
Upload 93 files
9016314 verified
raw
history blame
185 Bytes
def get_model(hps):
if hps.model == 'pc_acset_vae':
from .pc_acset_vae import ACSetVAE
model = ACSetVAE(hps)
else:
raise ValueError()
return model