QUEST2179

Thanks, I will try your suggestion.

Another error on save model. with open("beta_cnn.json", "w") as fout: json.dump(solver.config_dict(), fout) raise TypeError(f'Object of type {o.__class__.__name__} ' TypeError: Object of type TruncateProtein is not JSON serializable solver.save("beta_cnn.pth") works though.

Just follow your BetaLactamase example on your webpage. def testPropertyPrediction(): from torchdrug import models model = models.ProteinCNN(input_dim=21, hidden_dims=[1024, 1024], kernel_size=5, padding=2, readout="max") from torchdrug import transforms truncate_transform = transforms.TruncateProtein(max_length=200, random=False)...

Sorry this fix doesn't work. I still get the same error.

sorry I didn't pay much attention to the exact error message, use your latest repo, the error message has changed to the following raise TypeError(f'Object of type {o.__class__.__name__} ' TypeError:...