大佬您好问一下resnet34的问题

[dddtyz]

我改进了您的resnet18，新添加了模块，效果还不错。但是我怎么吧backbone改成resnet34呢？只是config里面把18换成34就自动换成34了吗。求解答

[cfzd]

@dddtyz
对的，改成34即可

Ultra-Fast-Lane-Detection/model/backbone.py

Lines 17 to 20 in 353df10

	if layers == '18':
	model = torchvision.models.resnet18(pretrained=pretrained)
	elif layers == '34':
	model = torchvision.models.resnet34(pretrained=pretrained)

[huijie0208]

请问你添加模块是加在哪个文件里呀