animl (version 3.2.0)

save_classifier: Save model state weights

Description

Save model state weights

Save model state weights

Usage

save_classifier(
  model,
  out_dir,
  epoch,
  stats,
  optimizer = NULL,
  scheduler = NULL
)
save_classifier(
  model,
  out_dir,
  epoch,
  stats,
  optimizer = NULL,
  scheduler = NULL
)

Value

None

None

Arguments

model

pytorch model

out_dir

directory to save model to

epoch

current training epoch

stats

performance metrics of current epoch

optimizer

pytorch optimizer (optional)

scheduler

pytorch scheduler (optional)

Examples

if (FALSE) save_classifier(model, 'models/', 10, list(acc = 0.85))
if (FALSE) save_classifier(model, 'models/', 10, list(acc = 0.85))