文件名称:neural-synthesizer
文件大小:33KB
文件格式:ZIP
更新时间:2024-04-04 17:29:37
Python
神经合成器 python modular_train.py \ --dataset mnist \ --model simple \ --n_parts 2 \ --loss xe \ --lr1 .1 \ --lr2 .1 \ --activation relu \ --optimizer sgd \ --weight_decay1 .0002 \ --weight_decay2 .0002 \ --seed 5 \ --print_freq 1 \ --n_epochs1 70 \ --n_epochs2 70 \ --hidden_objective srs_upper_tri_alignment \ --loglevel info \ --n_classes 10 \ --augment_data False \ --in_channels 1 \ --batch_size 128
【文件预览】:
neural-synthesizer-main
----modular_train.py(5KB)
----kernet()
--------models()
--------trainers()
--------__init__.py(243B)
--------parsers()
--------utils()
--------datasets()
----requirements.txt(248B)
----setup.py(625B)
----.gitignore(3KB)
----conda-requirements.yml(213B)
----README.md(468B)