 Using the pre-trained models You can test and use the preetrained models using the following codes which are available in the test.py: The pretrained student model are also located in "models/students".    ```   cnn = CNNModel()         model = cnn.get_model(arch=arch, input_tensor=None, output_len=self.output_len)          model.load_weights(weight_fname)          img = None 