Artificial Intelligence, Machine Learning, Deep Learning, Image Classification, Heartbeat.
param in self.pretrain_model.parameters(): param.requires_grad = False self.pretrain_model.fc = nn.Sequential( nn.Linear(self.pretrain_model.fc.in_features, 1024), nn.ReLU(), nn.Dropout(), nn.Linear(1024,self.num_classes) ) # To run data through the model def forward(self, input):