RuntimeError: masked_fill_ only supports boolean masks, but got mask with dtype unsigned char

[cool-RR]

When running the Colab:

old, tau=1.0, lambda=1.0
bs 64
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
[<ipython-input-9-168f548fe4ca>](https://localhost:8080/#) in <cell line: 3>()
      1 # Check the correlation of different estimators with the true derivatives as a function of batch size
      2 
----> 3 run_corr_exp()
      4 plot_corr_exp()

2 frames
[<ipython-input-2-9f6565e0a71e>](https://localhost:8080/#) in make_objective(batch_data, v_fn, tau, dice_lambda, use_dice, gamma_weighted)
     22         batch_r, batch_dones, batch_mask, ep_returns) = batch_data
     23     empty_mask = (1 - batch_mask[:,:-1]).type(torch.ByteTensor)
---> 24     batch_pi_taken[empty_mask] = 1.0
     25     log_pi = torch.log(batch_pi_taken)
     26     batch_values = v_fn(batch_states).detach()

RuntimeError: masked_fill_ only supports boolean masks, but got mask with dtype unsigned char

Maybe you've used an older version of some dependency. If so, I suggest you'd pin the colab to the versions you used.