o zi @sZdZddlZddlZddlZddlZddlZddlZddlmZddl Z GdddZ dS)zS Useful class for Experiment tracking, and ensuring code is saved alongside files. N)Pathc@s\eZdZdZ  ddedefddZdd Zd d Zed d Z ddfde j fddZ dS) ExperimentaZThis class contains utilities for managing experiments. It is a context manager, that when you enter it, changes your directory to a specified experiment folder (which optionally can have an automatically generated experiment name, or a specified one), and changes the CUDA device used to the specified device (or devices). Parameters ---------- exp_directory : str Folder where all experiments are saved, by default "runs/". exp_name : str, optional Name of the experiment, by default uses the current time, date, and hostname to save. runs/N exp_directoryexp_namecCsd|dur|}t||}|jddd||_||_ttd d |_ td |_ dS)NTparentsexist_okz+git ls-tree --full-tree --name-only -r HEADzutf-8.)generate_exp_namermkdirexp_dirr subprocess check_outputshlexsplitdecode splitlinesgit_tracked_filesabsoluteparent_directory)selfrrr rL/home/ubuntu/.local/lib/python3.10/site-packages/audiotools/ml/experiment.py__init__!s zExperiment.__init__cCst|_t|j|SN)osgetcwdprev_dirchdirr )rrrr __enter__6s  zExperiment.__enter__cCst|jdSr)rrr)rexc_type exc_value tracebackrrr__exit__;szExperiment.__exit__cCs&tjd}|dt}|S)zGenerates a random experiment name based on the date and a randomly generated adjective-noun tuple. Returns ------- str Randomly generated experiment name. z%y%m%d-)datetimenowstrftime randomnameget_name)datenamerrrr >s zExperiment.generate_exp_namecCsdS)NTr)frrrLszExperiment. filter_fncCs>|jD]}||rt|jjdddt|j||qdS)aCaptures a full snapshot of all the files tracked by git at the time the experiment is run. It also captures the diff against the committed code as a separate file. Parameters ---------- filter_fn : typing.Callable, optional Function that can be used to exclude some files from the snapshot, by default accepts all files TrN)rrparentr shutilcopyfiler)rr/r-rrrsnapshotLs zExperiment.snapshot)rN) __name__ __module__ __qualname____doc__strrr r$ staticmethodr typingCallabler3rrrrrs   r) r7r&rrr1rr:pathlibrr)rrrrrs