Error with CMHE model

[boiarsky18]

Hi, I am getting the following error when calling the fit function on the DeepCoxMixturesHeterogenousEffects model using default batch_size = 100. Can you please clarify if there is an error in the package? Thanks!

Error: expand(torch.FloatTensor{[100, 100]}, size=[100]): the number of sizes provided (1) must be greater or equal to the number of dimensions in the tensor (2)