Method gradinator

train.py:179–181

(x)

    def prepare_data(self, datum, devices:list=None, allocation:list=None):

        def gradinator(x):
            x.requires_grad = False
            return x
        if devices is None:
            devices = ['cuda:0']
        if allocation is None: