add dimension to target tensor to prevent valueerror

aspuru-guzik-group / GA

Code for the paper: Augmenting genetic algorithms with deep neural networks for exploring the chemical space

98 stars 36 forks source link

add dimension to target tensor to prevent valueerror #3

Closed seyonechithrananda closed 3 years ago

seyonechithrananda commented 3 years ago

When using newer torch versions, the existing code throws the following error:

ValueError: Using a target size (torch.Size([1000])) that is different to the input size (torch.Size([1000, 1])) is deprecated. Please ensure they have the same size.

By adding an additional dimension to the target tensor this issue is resolved.

@akshat998