train Namespace Reference

Functions
	save_benchmark (duration, final_loss, epoch_history)
	train ()

Variables
int	PLUGIN_PARAM_COUNT = 9
int	EPOCHS = 100
int	LR = 1e-4
int	BATCH_SIZE = 8
str	DATASET_PATH = "../data/processed/presets.json"
str	BENCHMARK_FILE = "../benchmarks/history.json"
str	MODEL_SAVE_PATH = "../saved_models/my_plugin_ai.pth"

Function Documentation

save_benchmark()

train.save_benchmark	(		duration,
			final_loss,
			epoch_history )

train()

train.train

(

)

Variable Documentation

BATCH_SIZE

int train.BATCH_SIZE = 8

BENCHMARK_FILE

str train.BENCHMARK_FILE = "../benchmarks/history.json"

DATASET_PATH

str train.DATASET_PATH = "../data/processed/presets.json"

EPOCHS

int train.EPOCHS = 100

LR

int train.LR = 1e-4

MODEL_SAVE_PATH

str train.MODEL_SAVE_PATH = "../saved_models/my_plugin_ai.pth"

PLUGIN_PARAM_COUNT

int train.PLUGIN_PARAM_COUNT = 9