path	# lines of code
models.py	227
setup_datasets.py	180
train_toy.py	46
datasets.py	282
plot_toy_scatter.py	170
train.py	141
environment.yaml	21
parse.py	478