Error: object has no attribute 'num_train_imgs' in version master, but not 0.6.0
I am running on Ubuntu 18, and trying to run the vq_vae.yaml config from this package:
https://github.com/AntixK/PyTorch-VAE
It runs fine under version 0.6.0 with python 3.7 but when I update to lighting-master (0.6.1)  I get this error:
AttributeError: 'VAEXperiment' object has no attribute 'num_train_imgs'
The relevant .py file is attached:
experiment.zip
Is there a change-log that would tell me what might be wrong since AntixK wrote this package for 0.6.0 and I'd like to use it with Apex, which isn't supported on 0.6.0

🐛 Bug

To Reproduce
Steps to reproduce the behavior:

Go to '...'
Run '....'
Scroll down to '....'
See error


Code sample

Expected behavior

Environment
Please copy and paste the output from our
environment collection script
(or fill out the checklist below manually).
You can get the script and run it with:
wget https://raw.githubusercontent.com/pytorch/pytorch/master/torch/utils/collect_env.py
# For security purposes, please check the contents of collect_env.py before running it.
python collect_env.py


PyTorch Version (e.g., 1.0):
OS (e.g., Linux):
How you installed PyTorch (conda, pip, source):
Build command you used (if compiling from source):
Python version:
CUDA/cuDNN version:
GPU models and configuration:
Any other relevant information:

Additional context