1068687f83 | ||
---|---|---|
.. | ||
data | ||
data2 | ||
data4 | ||
pixel_cnn_pp | ||
README.MD | ||
batch_test_script_leftpart.sh | ||
batch_test_script_mainbody.sh | ||
cifar_input.py | ||
example.sh | ||
monitor.py | ||
monitor_GivenCLM.py | ||
monitor_checkleft.py | ||
resnet_model_basic.py | ||
test_grab_train_loss.py | ||
train.py | ||
worker_I2L.py | ||
worker_L2I.py |
README.MD
Thanks a lot for your interests to our work. I quickly wrap up a multi-gpu version code. (Note that in the submission phase, we use a 1gpu version code. Please let me know if anything we can improve)
Training demo code: example.sh
Inference demo code: batch_test_script_mainbody.sh
The data and the checkpoint are available at: https://www.dropbox.com/sh/fpnvtcmyj4mul2s/AAB4wvsxoS8pf7ExnZYe4VV1a?dl=0
You need to download them and put them in the working dir. An example is in ``example.sh''