train.py File Reference

Go to the source code of this file.

Classes
class	train.AlignLayer

Namespaces
namespace	train

Functions
TensorDataset	train.load_dataset (str path)
tuple[float, float, float]	train.train_one (TensorDataset dataset, str det_id, int epochs, float lr, int batch, str out_dir, torch.device device)
	train.main ()

Variables
str	train.DEFAULT_GLOB = "/home/igor/DATA/bmn/nn/data_det*.csv"
int	train.DEFAULT_EPOCHS = 100
int	train.DEFAULT_LR = 1e-3
int	train.DEFAULT_BATCH = 256
str	train.DEFAULT_OUT_DIR = "models"
	train.iX
	train.iY
	train.iZ
	train.iXA
	train.iXB
	train.iYA
	train.iYB
list	train.INPUT_COLS = [iZ, iXA, iYA, iXB, iYB]
list	train.OUTPUT_COLS = [iX, iY]