required for better transfer learning implementation
required for better transfer learning implementation