thanks SuccessfulKoala55 , sorry that I can’t share the whole code but here is my main part
` import faulthandler
import logging

import torch
from argparse import ArgumentParser
from pytorch_lightning.callbacks import ModelCheckpoint

...

import os

from trains import Task
model_snapshots_path = 'mnt/trains'
if not os.path.exists(model_snapshots_path):
os.makedirs(model_snapshots_path)

task = Task.init(project_name='xl_trainer',
task_name='extending automagical Trains xl trianer',
output_uri=model_snapshots_path)

logger = task.get_logger()

def main(args):

 ` ` ('Got args: %s', args)

model = XLModelFactory().create_or_load(args)

...

if name == 'main':
parent_parser = ArgumentParser(add_help=False)
main(ResnetCIFAR10Model.add_model_specific_args(parent_parser).parse_args(nested=True)) `