How to save optimizer?

[sammyne]

Just to want to find a way to save the optimizer and load it somewhere else.

Trying to save the output (e.g. of type HashMap<ParamId, AdaptorRecord<..>>) of Optimizer::to_record using NamedMpkFileRecorder doesn't work with errors as

the trait bound `ParamId: Serialize` is not satisfied

[laggui]

You should be able to save the optimizer state/record using optimizer.to_record() and a file recorder, e.g.

recorder.record(optim.to_record(), save_path).unwrap();

This is pretty much what we do in burn-train:

burn/crates/burn-train/src/checkpoint/file.rs

Lines 46 to 55 in 6efc6fe

	fn save(&self, epoch: usize, record: R) -> Result<(), CheckpointerError> {
	let file_path = self.path_for_epoch(epoch);
	log::info!("Saving checkpoint {} to {}", epoch, file_path.display());
	self.recorder
	.record(record, file_path)
	.map_err(CheckpointerError::RecorderError)?;
	Ok(())
	}

[sammyne]

Confirm it works~