Migrating from Ignite"

knitr::opts_chunk$set(echo = TRUE,eval = FALSE,echo = T)

_

Intro

Ignite is a high-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

Ignite with fastai

library(fastai)
library(magrittr)

data = Data_Loaders(get_data_loaders(64, 128))$cuda()

nn = nn()
opt_func = partial(SGD, momentum=0.5)
learn = Learner(data, Net(), loss_func = nn$NLLLoss(), opt_func = opt_func, metrics = accuracy)
learn %>% fit_one_cycle(1, 0.01)
epoch     train_loss  valid_loss  accuracy  time    
0         1.084753    0.908347    0.826600  00:13  


Try the fastai package in your browser

Any scripts or data that you put into this service are public.

fastai documentation built on June 22, 2024, 11:15 a.m.