まぐらぼ

日々の雑記を書いています。

2020-02-28から1日間の記事一覧

jetbot livedemo

import torchvision import torchmodel = torchvision.models.resnet18(pretrained=False) model.fc = torch.nn.Linear(512, 2)model.load_state_dict(torch.load('best_steering_model_xy.pth')) device = torch.device('cuda') model = model.to(device) m…

jetbot train.py

import torchvision import torchmodel = torchvision.models.resnet18(pretrained=False) model.fc = torch.nn.Linear(512, 2)model.load_state_dict(torch.load('best_steering_model_xy.pth')) device = torch.device('cuda') model = model.to(device) m…