class_or_func: CrossEntropyLoss module_path: torch.nn args: [] kwargs: {}