Thanks AgitatedDove14 , I would love to but I’m afraid of can’t put time on it as abit busy with couple projects at this point, I will let you know when I got free to helping you on this. 🙏

AgitatedDove14 how far you are to make it presentable? what’s needs to be done , maybe I can put some efforts on it

Thanks AgitatedDove14 yes task.connect worked. about jsonargparse not sure how popular it is but in our case regular argparse didn’t work and we find out jsonargparse is working perfect. I guess will be beneficial to support that as well.

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',...

AgitatedDove14 thanks but I’m not using argparse , I’m using jsonargparse
you can see the simplified snippet here
` from jsonargparse import ArgumentParser
from trains import Task
model_snapshots_path = 'mnt/trains'
task = Task.init(project_name='examples', task_name='test argparser', output_uri=model_snapshots_path)
logger = task.get_logger()
def main(args):
print('Got args: %s' % args)
if name == 'main':
parent_parser = ArgumentParser(add_help=False)
parent...

oh great 🙏