Spaces:
Sleeping
Sleeping
File size: 185 Bytes
9016314 |
1 2 3 4 5 6 7 8 9 10 |
def get_model(hps):
if hps.model == 'pc_acset_vae':
from .pc_acset_vae import ACSetVAE
model = ACSetVAE(hps)
else:
raise ValueError()
return model
|