search

avik-pal / Wandb.jl

Unofficial Julia bindings for logging experiments to wandb.ai
https://avik-pal.github.io/Wandb.jl/stable/
MIT License
82 stars 11 forks source link

Logging the model's parameters - initialize ´wblogger´ #2

Closed saraalrawi closed 2 years ago

saraalrawi commented 2 years ago

Hello everyone,

Thanks for the nice work. Could you please tell me where is ´wblogger´ defined in the following line?

´log(wblogger, cpu; parameters = ps, gradients = gs, commit = false)´

Thanks in advance!

avik-pal commented 2 years ago

Thanks for pointing it out. wblogger should be something like WandbLogger( project = "Wandb.jl", name = "fluxjl-integration-$(now())", config = Dict( "learning_rate" => 3e-4, "batchsize" => 256, "epochs" => 100, "dataset" => "MNIST", "use_cuda" => true, ), ). But I removed that particular function dispatch in the current release (I haven't updated the docs). The current way to do it would be to log the histograms manually.