Spaces:

Hanokalure
/

human-activity-backend

Runtime error

Deploy Human Activity Recognition backend to Hugging Face Spaces

062cf9b about 1 year ago

243 Bytes

import torch.nn as nn
from torchvision import models

def get_model(num_classes=101):  # 👈 UCF101 has 101 classes
    model = models.video.r3d_18(pretrained=True)
    model.fc = nn.Linear(model.fc.in_features, num_classes)
    return model

	import torch.nn as nn
	from torchvision import models

	def get_model(num_classes=101): # 👈 UCF101 has 101 classes
	model = models.video.r3d_18(pretrained=True)
	model.fc = nn.Linear(model.fc.in_features, num_classes)
	return model