Modify the Trainer
Copy nn to img inside thread function.