Imported from https://github.com/stanford-cs348k/asst2-mobilenet