Breadcrumbs VAC_cslr / main.py Latest commit Yifan Wang latest_update a950734· Oct 11, 2022 HistoryHistory File metadata and controls Code Blame 56 lines (48 loc) · 2.95 KB Raw import torch.cuda import pickle from model import * from utils.decode import Decoder from utils.optimizer ...
class CSLR(nn.Module): def __init__(self, spatio_dim, num_classes, hidden_dim, decoder): super(CSLR, self).__init__() # backbone: resnet18 self.conv2d = torch.hub.load('pytorch/vision:v0.10.0', 'resnet18', pretrained=True) self.conv2d_fc = nn.Linear(1000, spatio_dim) self...
line 211, in processor.start() File "/home/user/VAC_CSLR-main/main.py", line 63, in start "dev", 6667, self.arg.work_dir, self.recoder, self.arg.evaluate_tool) File "/home/user/VAC_CSLR-main/seq_scripts.py", line 56, in seq_eval ret_dict =...