update docs notebooks

Author: lorenzoh

Manifest.toml

@@ -228,11 +228,9 @@ version = "1.6.0"
 
 [[DataAugmentation]]
 deps = ["ColorBlendModes", "CoordinateTransformations", "ImageDraw", "ImageTransformations", "Images", "Interpolations", "LinearAlgebra", "Parameters", "Rotations", "Setfield", "StaticArrays", "Statistics"]
-git-tree-sha1 = "2e9b277d0e5458c84e0c310cdfb2d591901dda4e"
-repo-rev = "master"
-repo-url = "https://github.com/lorenzoh/DataAugmentation.jl"
+git-tree-sha1 = "8138afd39eb19c8b47e094fb7a18fb8f752cc3b3"
 uuid = "88a5189c-e7ff-4f85-ac6b-e6158070f02e"
-version = "0.1.0"
+version = "0.1.1"
 
 [[DataDeps]]
 deps = ["BinaryProvider", "HTTP", "Libdl", "Reexport", "SHA", "p7zip_jll"]
@@ -527,9 +525,9 @@ version = "0.8.20"
 
 [[ImageDistances]]
 deps = ["ColorVectorSpace", "Distances", "ImageCore", "ImageMorphology", "LinearAlgebra", "Statistics"]
-git-tree-sha1 = "159e24b4313d9197eef900e97fbd7365986f2844"
+git-tree-sha1 = "622fdcb6ec80aee6f6620f3788ea269702a24545"
 uuid = "51556ac3-7006-55f5-8cb3-34580c88182d"
-version = "0.2.10"
+version = "0.2.11"
 
 [[ImageDraw]]
 deps = ["Distances", "ImageCore", "LinearAlgebra"]
@@ -538,10 +536,10 @@ uuid = "4381153b-2b60-58ae-a1ba-fd683676385f"
 version = "0.2.4"
 
 [[ImageFiltering]]
-deps = ["CatIndices", "ColorVectorSpace", "ComputationalResources", "DataStructures", "FFTViews", "FFTW", "ImageCore", "ImageMetadata", "LinearAlgebra", "OffsetArrays", "Requires", "SparseArrays", "StaticArrays", "Statistics", "TiledIteration"]
-git-tree-sha1 = "f82a52fa2e684d4ed69028b16188852ff94b3f75"
+deps = ["CatIndices", "ColorVectorSpace", "ComputationalResources", "DataStructures", "FFTViews", "FFTW", "ImageCore", "LinearAlgebra", "OffsetArrays", "Requires", "SparseArrays", "StaticArrays", "Statistics", "TiledIteration"]
+git-tree-sha1 = "81b493f57c67a8dff822e43cb2abefece7b52131"
 uuid = "6a3955dd-da59-5b1f-98d4-e7296123deb5"
-version = "0.6.19"
+version = "0.6.20"
 
 [[ImageIO]]
 deps = ["FileIO", "Netpbm", "PNGFiles"]

docs/make.jl

@@ -11,6 +11,7 @@ download_artifact(
 )
 
 Publish.Themes.default() = artifact"flux-theme"
+ENV["DATADEPS_ALWAYS_ACCEPT"] = "true"
 
 p = Publish.Project(FastAI)
 

docs/notebooks/imageclassification.ipynb docs/notebooks/finetune.ipynb

@@ -6,4 +6,10 @@ FastAI.jl is not registered yet since it depends on some unregistered packages (
 ] clone https://github.com/lorenzoh/FastAI.jl
 ] activate FastAI
 ] instantiate
+```
+
+FastAI.jl also defines [Makie.jl] plotting recipes to visualize data. If you want to use them, you'll have to install and one of the Makie.jl backends [CairoMakie.jl](https://github.com/JuliaPlots/CairoMakie.jl), [GLMakie.jl](https://github.com/JuliaPlots/GLMakie.jl) or [WGLMakie.jl](https://github.com/JuliaPlots/WGLMakie.jl). For example:
+
+```julia
+] add CairoMakie
 ```

docs/notebooks/fitonecycle.ipynb

@@ -73,3 +73,28 @@ function FluxTraining.fitepochphase!(
 
     phase.result = LRFinderResult(losses, lrs)
 end
+
+
+
+function plotlrfind(lrs, losses)
+    f = Figure(resolution = (700, 400))
+    f[1, 1] = ax = Axis(f)
+
+    xtickvalues = (10.) .^(-10:10)
+    ax.xlabel = "Learning rate"
+    ax.xticks = log.(xtickvalues)
+    ax.xtickformat[] = vals -> string.(xtickvalues)
+    ax.xticklabelsize = 10.
+
+    ax.ylabel = "Loss"
+    ax.yticks = []
+    ax.ytickformat[] = vals -> string.(log.(vals))
+    ax.yticklabelsize = 10.
+
+
+    AbstractPlotting.lines!(ax, log.(lrs), log.(losses), axis = (xticks = LinearTicks(10),))
+    return f
+end
+
+plotlrfind(phase::LRFinderPhase) = plotlrfind(phase.result)
+plotlrfind(lrresult::LRFinderResult) = plotlrfind(lrresult.lrs, lrresult.losses)

docs/notebooks/lrfind.ipynb

@@ -16,7 +16,7 @@
 # How To
 
 - [Train a model from scratch](docs/notebooks/fitonecycle.ipynb)
-- [Finetune a pretrained model](docs/notebooks/fitonecycle.ipynb)
+- [Finetune a pretrained model](docs/notebooks/finetune.ipynb)
 - [Find a good learning rate](docs/notebooks/lrfind.ipynb)
 
 # Reference