Accelerate ⚡ your ML research using pre-built Deep Learning Models with Lux
using Pkg
Pkg.add("Boltz")using Boltz, Lux, Metalhead
model, ps, st = resnet(:alexnet; pretrained=true)CUDA is not loaded by default. To use GPUs follow Lux.jl documentation.
We have moved some dependencies into weak dependencies. This means that you will have to manually load them for certain features to be available.
- To load Flux & Metalhead models, do
using Metalhead.