Siamese-pytorch
Siamese-pytorch copied to clipboard
这是一个孪生神经网络(Siamese network)的库,可进行图片的相似性比较。
怎么对训练出来的模型转onnx,然后简化!!最主要还是简化,原模型转onnx在cpu耗时太长,需要简化。不知道怎么简化,用torch的简化会报错
Traceback (most recent call last): File "train.py", line 344, in fit_one_epoch(model_train, model, loss, loss_history, optimizer, epoch, epoch_step, epoch_step_val, gen, gen_val, Epoch, Cuda, fp16, scaler, save_period, save_dir, local_rank) File "/home/stable/dianxuan/code/Siamese-pytorch/utils/utils_fit.py", line...
预测速度有些慢
按照说明的那样下载好模型 运行predict.py 最后能出结果但是特别慢差不多要20分钟
很多时候都会出现这个情况,如果把训练的随机种子换一下有可能就会迅速下降。 出现这种情况的原因是训练集少了吗?还是训练集少了?还是网络有问题? 大图和小图已经一共有13000多张了。
为什么训练后的trainloss始终比valloss大,trainacc始终比valacc小,没有改过代码