model/stash/14062024_0910/log.txt

----------------
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