search

lulab / OligoFormer

GNU General Public License v3.0
14 stars 3 forks source link

Oligo model error #3

Open zhendejuzi opened 4 months ago

zhendejuzi commented 4 months ago

python scripts/main.py --cuda 0 --learning_rate 0.0001 --batch_size 16 --epoch 100 --early_stopping 30

Traceback (most recent call last): File "scripts/main.py", line 62, in main() File "scripts/main.py", line 59, in main train_single(Args) if Args.single else train(Args) File "/data/wusijie/software/OligoFormer/scripts/train.py", line 179, in train OFmodel = Oligo(vocab_size = Args.vocab_size, embedding_dim = Args.embedding_dim, lstm_dim = Args.lstm_dim, n_head = Args.n_head, n_layers = Args.n_layers, lm1 = Args.lm1, lm2 = Args.lm2).to(device) File "model.py", line 277, in model.Oligo.init TypeError: init() got an unexpected keyword argument 'lm1'

byl18 commented 4 months ago

Since the paper was not submitted at that time, the model part was set to model.so, which has been updated to Model.py now.

byl18 commented 4 months ago

The Docker image is easier for users to get started with OligoFormer without worrying about dependencies and environment configuration.