Augment set to true causes RuntimeError

[matigekunstintelligentie]

When I set augment to true in the config file I get a RuntimeError on lines 66 and 49 of stylegan2/op/fused_act.py

RuntimeError: input must be contiguous

Unless I make the input contiguous.

gradgrad_out = fused.fused_bias_act( gradgrad_input.contiguous(), gradgrad_bias, out, 3, 1, ctx.negative_slope, ctx.scale )

out = fused.fused_bias_act(input.contiguous(), bias, empty, 3, 0, negative_slope, scale)