FlOp.jl:使用Optim.jl例程训练Flux.jl模型的函数

FlOp.jl（Flux + Optim） FlOp软件包包含一些功能，这些功能使人们可以使用带有或不带有GPU设施的Optim.jl中的优化例程来训练Flux.jl模型。 版本号 Juliav1.6.0 CUDA.jl v2.6.2 Flux.jl v0.12.0 Zygote.jl v0.6.7 Optim.jl v1.3.0 RecursiveArrayTools v2.11.0 导出功能 set_vars_methods set_vars_methods (loss, pars) set_vars_methods函数用于初始化FlOp.jl模块中的全局变量。 pars变量的类型必须为Flux.Params ， loss函数必须为指向所需成本函数的匿名函数，例如 loss = () -> Flux . Losses . mse ( network (x), y) to