Loading a tranied model from a .pth file via model.load() seems to work. A problem arises when sampling new data from this model with the ODE solver in FlowPipeLine. The following error arises in the return of the TorchDynWrapper:
TypeError: '_IncompatibleKeys' object is not callable
Loading a tranied model from a .pth file via model.load() seems to work. A problem arises when sampling new data from this model with the ODE solver in FlowPipeLine. The following error arises in the return of the TorchDynWrapper:
TypeError: '_IncompatibleKeys' object is not callable