Julia implementation of greenlearning
Original implementation written in Python
git clone git@github.com:yonesuke/GreenLearning.jl.git
cd GreenLearning.jl; git submodule update --init --recursive
To train neural network, run the following.
julia train.jl <EQUATION NAME> <EPOCHS>
For example, if you want to learn the green function of the Laplace equation with 100 epochs,
this command julia train.jl laplace 100
will do.
Trained model is saved to chekpoints_<EQUATION_NAME>_<DATE>/model_<Nth_EPOCH>.bson
for every five epochs.
Load the trained model with the following.
julia> using BSON: @load
julia> @load "chekpoints_<EQUATION_NAME>_<DATE>/model_<Nth EPOCH>.bson" G_network U_hom_network opt