model/train.py [909:916]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            if args.fp16:
                grad_norm = torch.nn.utils.clip_grad_norm_(
                    amp.master_params(optimizer), cfg.TRAIN.clip
                )
            else:
                grad_norm = torch.nn.utils.clip_grad_norm_(
                    model.module.generator.parameters(), cfg.TRAIN.clip
                )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



model/train.py [1080:1087]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            if args.fp16:
                grad_norm = torch.nn.utils.clip_grad_norm_(
                    amp.master_params(optimizer), cfg.TRAIN.clip
                )
            else:
                grad_norm = torch.nn.utils.clip_grad_norm_(
                    model.module.generator.parameters(), cfg.TRAIN.clip
                )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -