Solved by Janine in post #4 
              
              
                Solved the problem. It turns out need to pass  clean_loader.dataset  instead of  clean_loader  to  DataLoader. Just would like to document it here in case someone else also get caught with the same mistake.