球球triplet成功吧
01测试时RuntimeError: CUDA out of memory
解决办法https://zhuanlan.zhihu.com/p/364332471
02预训练模型加载到更改后的网络中
load_state_dict方法有一个输入:strict,如果该参数为True,就表示网络结构的层信息要和预训练模型的层信息严格相等,反之亦然,该参数默认是True。那么什么时候会用到False呢?就是当你只想用预训练网络初始化你的网络的部分层参数或者说你的预训练网络的层信息和你要被初始化的网络的层信息不完全一致,那样就只会初始化层信息相同的层。
03报错:RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)
batch_size改小就好啦