Hugging Face's logo

Spaces:
libokj
/
DeepSEQreen_fast_build
Running on CPU Upgrade

App Files Community
DeepSEQreen_fast_build
File size: 932 Bytes 
416833a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
# @package _global_
defaults:
  - override /data/drug_featurizer: mol_features
  - override /data/protein_featurizer: tokenizer

model:
  predictor:
    _target_: deepscreen.models.predictors.transformer_cpi_2.TransformerCPI2

    encoder:
      _target_: deepscreen.models.predictors.transformer_cpi_2.Encoder
      # /model/protein_encoder@pretrain: tape_bert
      n_layers: 3
      pretrain:
        _target_: tape.ProteinBertModel.from_pretrained
        pretrained_model_name_or_path: resources/models/tape/bert-base/ # bert-base

    decoder:
      _target_: deepscreen.models.predictors.transformer_cpi_2.Decoder
      n_layers: 3
      dropout: 0.1

data:
  batch_size: 16
  collator:
    automatic_padding: True

  protein_featurizer:
    tokenizer:
      _target_: tape.TAPETokenizer.from_pretrained
      vocab: iupac

#  collator:
#    _target_: deepscreen.models.predictors.transformer_cpi_2.pack
#    _partial_: true