kochbj/Deep-Learning-for-Causal-Inference

Tarnet short tutorial error

xiaogangzhu opened this issue · 1 comments

xiaogangzhu commented

Hi,

When i am running first tutorial of TarNet on colad, got the training loss = nan.
Screenshot from 2023-05-15 14-17-32

kochbj commented

Hi,

I fixed the TF version at 2.8. Hopefully that solves the issue for you. Please let me know if you find other bugs!

Warmly,
Bernie