anjikum's picture
made changes made based on checkpoint model
2583941 verified