SoloWayG
/

Molecule_transformer

Model card Files Files and versions

Molecule_transformer / GAN_weights /train_GAN_BTK_affinity_prediction

61.6 MB

Ctrl+K

Ctrl+K

1 contributor

History: 1 commit

This model has 16 files scanned as unsafe.

SoloWayG's picture

Add GAN model for BTK_affinity_prediction case

716c9aa verified 6 months ago

gan_weights.pkl
Detected Pickle imports (21)
- "builtins.dict",
- "torch.nn.modules.activation.ReLU",
- "torch.nn.modules.activation.ELU",
- "torch.nn.modules.rnn.LSTMCell",
- "torch.optim.adam.Adam",
- "infrastructure.generative_models.GAN.gan_lstm_refactoring.scripts.layers.RecurrentDiscriminator",
- "torch.storage._load_from_bytes",
- "infrastructure.generative_models.GAN.gan_lstm_refactoring.scripts.utils.LstmSeq2SeqEncoder",
- "torch.nn.modules.sparse.Embedding",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.rnn.LSTM",
- "infrastructure.generative_models.GAN.gan_lstm_refactoring.scripts.layers.Generator",
- "infrastructure.generative_models.GAN.gan_lstm_refactoring.scripts.model.MolGen",
- "torch.nn.modules.activation.Sigmoid",
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict",
- "torch.nn.modules.dropout.Dropout",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.container.Sequential",
- "collections.defaultdict",
- "infrastructure.generative_models.GAN.gan_lstm_refactoring.scripts.tokenizer.Tokenizer"
How to fix it?
61.6 MB
xet

Add GAN model for BTK_affinity_prediction case 6 months ago