karpathy / deep-vector-quantization

VQVAEs, GumbelSoftmaxes and friends
MIT License
533 stars 43 forks source link

Move inmap to data loader #8

Open crizCraig opened 3 years ago

crizCraig commented 3 years ago

Hey Andrej! Cool repo :+1: This is a small change that effectually computes model(x - 0.5) in visualize.ipynb. Otherwise model(x) - 0.5 looks like this before_mv_inmap After the change we get after_mv_inmap And since we always want to normalize (e.g. before training, validating, or visualizing), I just moved the normalization/inmapping back to the data loader.