Hugging Face's logo

Spaces:
mschuh
/
BarlowDTI
Running on Zero

App Files Community
1
BarlowDTI
File size: 1,052 Bytes 
84bfd88
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
37
38
39
40
41
42
 
----------------
Run description: Manual param optim
----------------
message: yes
path: all_drugbank_smiles_sequence_prost_preprocessor.pkl
load_preprocessor: True
radius: 2
n_bits: 1024
num_workers: 64
enc_n_neurons: 4096
enc_n_layers: 3
proj_n_neurons: 2048
proj_n_layers: 1
embedding_dim: 512
act_function: relu
aa_emb_size: 1024
loss_weight: 0.005
batch_size: 4096
epochs: 250
optimizer: adamw
learning_rate: 0.0003
beta_1: 0.9
beta_2: 0.999
weight_decay: 5e-05
step_size: 10
gamma: 0.1
include_negatives: False
hyperparameter_tuning: False
val_split: 0.1
aa_embedding: prost_t5
model_type: barlow_twins
device: cuda:0
msg: Manual param optim
start: 1718356109.3235965
data: <preprocessor.Preprocessor object at 0x72f2d495eb10>
train: <torch.utils.data.dataloader.DataLoader object at 0x72f2d3a66d50>
val: <torch.utils.data.dataloader.DataLoader object at 0x72f2d480e7b0>
file: <_io.BufferedReader name='all_drugbank_smiles_sequence_prost_preprocessor.pkl'>
t_preprocessing: 0
model: <barlow_twins.BarlowTwins object at 0x72f2d7652540>
t_model: 1