zzk1212 issues

Repositories
Issues
Comments

Results 1 issues of


                                            zzk1212

RuntimeError: setStorage: sizes [1, 8, 3, 1, 1, 1034], strides [8288, 1036, 1036, 1, 1036, 1], storage offset 0, and itemsize 4 requiring a storage size of 41432 are out of bounds for storage of size 33152

有人碰到过这个错误嘛？使用了serial也不管用。显存是够的应该，下面是代码： from sklearn.metrics import roc_auc_score from opacus.grad_sample import GradSampleModule from opacus import PrivacyEngine from utils import adjust_learning_rate class LocalUpdate(object): def __init__(self, idx=0,sample_len=2): self.loss_func = SmoothBinaryCrossEntropyWithLogitsLoss(smooth_factor=smooth_factor).to(config['device']) self.args = args self.times =...