issue with self.rho = nn.ParameterDict(self.rho)

[deanslee]

Hi all, I was running through the code below in the provided tutorial and came across an error.

model = Spectra.est_spectra(adata=adata, gene_set_dictionary=annotations, use_highly_variable=True, cell_type_key="cell_type_annotations", use_weights=True, lam=0.1, delta=0.001, kappa=None, rho=0.001, use_cell_types=True, n_top_vals=50, label_factors=True, overlap_threshold=0.2, clean_gs=True, min_gs_num=3, num_epochs=2)

When rho=None, est_spectra runs just fine.
The same error with rho=0.001 also happens with kappa=0.001.

[deanslee]

I might be wrong, but my current understanding is that the issue with nn.ParameterDict(self.rho) is that self.rho needs to be two dimensions.

This is my attempt to reproduce the error:

Then running self_rho = nn.ParameterDict(self_rho) reproduces

TypeError: cannot assign 'torch.DoubleTensor' object to parameter 'B_GC' (torch.nn.Parameter or None required)

[deanslee]

Could this issue be a result of this commit?