Error with CMHE model - Githubissues

autonlab / auton-survival

Auton Survival - an open source package for Regression, Counterfactual Estimation, Evaluation and Phenotyping with Censored Time-to-Events

http://autonlab.github.io/auton-survival

MIT License

315 stars 74 forks source link

Error with CMHE model #140

Open boiarsky18 opened 7 months ago

boiarsky18 commented 7 months ago

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)