mirror of
https://github.com/curioustorvald/Terrarum-sans-bitmap.git
synced 2026-06-09 23:44:04 +09:00
Autokem: CNN-based glyph labeller for Keming Machine
This commit is contained in:
16
Autokem/safetensor.h
Normal file
16
Autokem/safetensor.h
Normal file
@@ -0,0 +1,16 @@
|
||||
#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
|
||||
Reference in New Issue
Block a user