I'm training a custom dataset in yolor. I successfully run its once but after some time, I cant manage to do it very well.
The first error I noticed is in the training part:
Traceback (most recent call last): File "train.py", line 537, in <module>
train(hyp, opt, device, tb_writer, wandb) File "train.py", line 80, in train
ckpt = torch.load(weights, map_location=device) # load checkpoint File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 595, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 764, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, '<'.
then i traced it and found that the pre trained weights didnt load correctly with this code:
%cd /content/yolor
!bash scripts/get_pretrain.sh
ann give me this error:
/content/yolor
awk: cannot open ./cookie (No such file or directory)
rm: cannot remove './cookie': No such file or directory
and that's the first and main thing I noticed comparing to what I've done, it should load the weight in there.
its just giving me a pre-trained files with some HTML code.
im using goolge colab btw