- A pytorch implementation of Attentive Neural Process.
- Simple code for generating samples with ANP.
- I will update the super-resolution experiments soon.
- Install python 3
- Install pytorch == 0.4.0
preprocess.pyincludes all preprocessing codes when you loads data.
module.pycontains all methods, including attention, linear and so on.
network.pycontains whole structure of network.
train.pyis for training ANP model.
generate.ipynbis for generating samples.
test samples after 50 epoch training with random context selection.
- Any comments for the codes are always welcome.