首页 > 解决方案 > 由于策略争论,使用 NueralPDE.jl 中的 PhysicsInformedNN() 时出现 MethodError

问题描述

我正在尝试通过阅读此处找到的文档教程来学习用于某些 PINN 工作的 NeuralPDE.jl 的语法。

使用 PhysicsInformedNN(chain, strategy) 函数构建 PINN 算法时,无论我尝试使用哪种策略,都会收到 MethodError。该教程建议使用 GridTraining(.05),我也尝试了 StochasticTraining(::Int64),但没有成功。

dim = 2 # number of dimensions
chain = FastChain(FastDense(dim,16,Flux.σ),FastDense(16,16,Flux.σ),FastDense(16,1))
dx = 0.05
discretization = PhysicsInformedNN(chain,GridTraining(dx))

错误:

MethodError: no method matching GridTraining(::Float64)

标签: juliapdeflux.jl

解决方案


这是一个包版本问题。更新到最新版本时,它已修复。有关更多信息,请参阅评论。


推荐阅读