1.RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same.
Code in linear_eval.py line 50 and 59 should add .to(device) :
model = get_backbone(args.backbone).to(args.device)
2.RuntimeError: main thread is not in main loop
Tcl_AsyncDelete: async handler deleted by the wrong thread
In plot_logger.py add:
import matplotlibmatplotlib.use('Agg')
before:
import matplotlib.pyplot as plt
PatrickHua
commented
3 years ago
1.RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same. Code in linear_eval.py line 50 and 59 should add
.to(device):model = get_backbone(args.backbone).to(args.device)2.RuntimeError: main thread is not in main loop Tcl_AsyncDelete: async handler deleted by the wrong thread In plot_logger.py add:
import matplotlibmatplotlib.use('Agg')before:import matplotlib.pyplot as plt