[GTSRB] Type error when loading loss weights

[opmusic]

When loading loss weights from GTSRB dataset, I found the default type of loss weights is np.float64, which will generate the following error:

RuntimeError: expected scalar type Float but found Double

And it may be fixed to add the following strict type conversion before line 182, trojanzoo/trojanzoo/models.py:

 if (isinstance(loss_weights,np.ndarray)):
     loss_weights=np.asarray(loss_weights).astype(np.float32)

[ain-soph]

Thanks for pointing it out. It seems this issue is introduced by new pytorch/numpy versions. (It still works at least last month.)

I'd prefer to keep loss_weights as double and transform to float when moving to torch.Tensor, because numpy default precision is double and pytorch precision is float.

I'll draft a new release very soon.

[ain-soph]

trojanzoo/trojanzoo/models.py

Line 182 in 7fbfb1b

self.criterion = self.define_criterion(weight=to_tensor(loss_weights, dtype=torch.float))