Error while using your weights as pretrained weights

[Deadpool5549]

Hi @gautamtata , I have used your yolov5s weights and code in colab notebook, but while running train.py or detect.py, I got the following error !python train.py --img 416 --batch 10 --epochs 10 --data '/content/data.yaml' --cfg ./models/custom_yolov5s.yaml --weights /content/best.pt --cache val: data=../data.yaml, weights=['/content/last.pt'], batch_size=32, imgsz=416, conf_thres=0.001, iou_thres=0.6, task=val, device=, workers=8, single_cls=False, augment=False, verbose=False, save_txt=False, save_hybrid=False, save_conf=False, save_json=False, project=runs/val, name=exp, exist_ok=False, half=True, dnn=False YOLOv5 🚀 v6.0-213-gd5966c9 torch 1.10.0+cu102 CUDA:0 (Tesla T4, 15110MiB)

Traceback (most recent call last): File "val.py", line 382, in main(opt) File "val.py", line 355, in main run(**vars(opt)) File "/usr/local/lib/python3.7/dist-packages/torch/autograd/grad_mode.py", line 28, in decorate_context return func(*args, **kwargs) File "val.py", line 139, in run model = DetectMultiBackend(weights, device=device, dnn=dnn, data=data) File "/content/yolov5/models/common.py", line 308, in init model = attempt_load(weights if isinstance(weights, list) else w, map_location=device) File "/content/yolov5/models/experimental.py", line 96, in attempt_load ckpt = torch.load(attempt_download(w), map_location=map_location) # load File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 600, in load with _open_zipfile_reader(opened_file) as opened_zipfile: File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 242, in init super(_open_zipfile_reader, self).init(torch._C.PyTorchFileReader(name_or_buffer)) RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory