使用PyTorch在基准数据集上培训SOTA体系结构 参数列表包括: 数据集(请参见下文) ptr表示火车的尺寸。 建筑(见下文) 优化器( sgd , adam ) lr和lr调度程序( cosineannealing , none ) 损失函数(多类crossentropy ,一类hinge ) 用alpha技巧在特征或懒惰机制中进行训练( featlazy变为1并改变alpha ) ... 例: python main.py --epochs 200 --save_best_net 1 --save_dynamics 0 --diffeo 0 --batch_si