search

meta-soul / MetaSpore

A unified end-to-end machine intelligence platform
Apache License 2.0
642 stars 118 forks source link

[algos][bug] Please specify the dimension of squeeze function in din model #209

Closed intelligencegear closed 2 years ago

intelligencegear commented 2 years ago

Please specify the dimension of squeeze function in din model, otherwise it may lead to a crash in model training.