lululxvi / deeponet

Learning nonlinear operators via DeepONet
Other
509 stars 139 forks source link

AttributeError: 'Triple' object has no attribute 'transform_inputs' #16

Closed cfd-ai closed 2 years ago

cfd-ai commented 2 years ago

Any idea how to resolve the following error? This is for the default problem = "ode" test case. Backend: tensorflow.compat.v1 TensorFlow version: 2.6.2 Thanks.

Restoring model from model/model.ckpt-37000.ckpt ...

Traceback (most recent call last): File "deeponet_pde.py", line 285, in main() File "deeponet_pde.py", line 281, in main run(problem, system, space, T, m, nn, net, lr, epochs, num_train, num_test) File "deeponet_pde.py", line 177, in run safe_test(model, data, X_test, y_test) File "/[HOME-PATH]/tests/deeponet/src/utils.py", line 56, in safe_test y_pred.append(model.predict(data.transform_inputs(X_add))) AttributeError: 'Triple' object has no attribute 'transform_inputs'

cfd-ai commented 2 years ago

If I understand correctly, I guess something like this would fix it:

    # Network
    nn = "DeepONet" #OpNN
    if nn == "DeepONet":
        y_pred.append(model.predict(X_add))
    else:
        y_pred.append(model.predict(data.transform_inputs(X_add)))
lululxvi commented 2 years ago

You are right.

Also see https://github.com/lululxvi/deeponet/issues/14#issuecomment-1025056619