AvlTree issues

Repositories
Issues
Comments

Results 2 issues of


                                            AvlTree

cudaCheckError() failed : invalid device function

有没有遇到相同问题的小伙伴可以解决一下

不能多卡训练，没有找到解决办法 ValueError: Caught ValueError in replica 0 on device 0.

Traceback (most recent call last): File "/home/avltree/Projects/Siamese-pytorch/train.py", line 202, in fit_one_epoch(net,loss,epoch,epoch_size,epoch_size_val,gen,gen_val,Freeze_Epoch,Cuda) File "/home/avltree/Projects/Siamese-pytorch/train.py", line 62, in fit_one_epoch outputs = net(images) File "/home/avltree/miniconda3/envs/siamese/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in __call__ result = self.forward(*input, **kwargs)...