Open joyousrabbit opened 4 years ago
Hello, it's great work! Really great.
However, the resume doen't work.
Besides, in training, the optimizer is created on model's cpu, however, the model is later put on gpu by model.cuda() in train.py, it will cause many issues later.
Could you please share the weight of your download? Because the current download link has expired
Hello, it's great work! Really great.
However, the resume doen't work.
Besides, in training, the optimizer is created on model's cpu, however, the model is later put on gpu by model.cuda() in train.py, it will cause many issues later.