train.py File Reference

Namespaces
namespace	train

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

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