mossr / WeightsAndBiasLogger.jl

Log to W&B from Julia

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

WeightsAndBiasLogger.jl

Travis CI Coveralls LICENSE
Build Status Coverage Status License

Installation

  • From REPL: ] add WeightsAndBiasLogger
  • By code: using Pkg; pkg"add WeightsAndBiasLogger"

Demo

using Logging, WeightsAndBiasLogger

args = (n_epochs=1_000, lr=1e-3)

logger = WBLogger(project="sample-project")

config!(logger, args)

with(logger) do
    loss = 0
    for i in 1:args.n_epochs
        loss += randn() * args.lr
        @info "train" i=i loss=loss
    end
end

About

Log to W&B from Julia

License:MIT License


Languages

Language:Julia 100.0%