AttributeError: module 'models2' has no attribute 'DemoLab'

[xtendai]

Running the LSTM baseline with lab / demographics gives the following error :

LSTM baseline with lab/demographics

python3 -u trainModel.py --i 2 --modelName DemoLab --dimLSTM 256 --p_dropOut 0.5 --batch_norm --flg_cuda --logInterval 1 --n_iter 2 --batchSizePos 8 --batchSizeNeg 8 --enc_len 30 --lr 0.001 --lr_decay3 10 --inputPath sampleData/ --flgSave --savePath DemoLab_Output/

AttributeError: module 'models2' has no attribute 'DemoLab'

[BerlyJIa]

I have same issue

[jingsliu]

Should be calling from enc_models, sorry about the error. Fixed and please try and let me know

[BerlyJIa]

I have test it today, the problem has been soveled. Thank you !

[Delta98]

It show cuda out of memory if I run the code. ( in anaconda)

[jingsliu]

That sounds like a configuration issue on your end, please try with a GPU with larger memory, or reduce the batch size.

…

On Thu, Nov 14, 2019 at 2:28 PM Delta98 ***@***.***> wrote: It show cuda out of memory if I run the code. ( in anaconda) — You are receiving this because you commented. Reply to this email directly, view it on GitHub <#1?email_source=notifications&email_token=AETMZYDLXOTRTIHLTCBVONDQTWRGHA5CNFSM4GKWE5G2YY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOEEDAFFY#issuecomment-554042007>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AETMZYGG4CM3REANC4BSBQTQTWRGHANCNFSM4GKWE5GQ> .