mirror of
https://github.com/curioustorvald/Terrarum-sans-bitmap.git
synced 2026-03-07 11:51:50 +09:00
17 lines
469 B
C
17 lines
469 B
C
#ifndef SAFETENSOR_H
|
|
#define SAFETENSOR_H
|
|
|
|
#include "nn.h"
|
|
|
|
/* Save network weights to .safetensors format.
|
|
metadata: optional string pairs (key,value,...,NULL) */
|
|
int safetensor_save(const char *path, Network *net, int total_samples, int epochs, float val_loss);
|
|
|
|
/* Load network weights from .safetensors file. */
|
|
int safetensor_load(const char *path, Network *net);
|
|
|
|
/* Print model stats from .safetensors file. */
|
|
int safetensor_stats(const char *path);
|
|
|
|
#endif
|