search

LambdaLabsML / examples

Deep Learning Examples
MIT License
805 stars 103 forks source link

The following error occurred when I ran main.py #52

Closed WhiteZz1 closed 1 year ago

WhiteZz1 commented 1 year ago 
old_state[k] = modify_weights(old_state[k], scale=scale, n=in_shape // 4 - 1)

TypeError: unsupported operand type(s) for //: 'torch.Size' and 'int'