파이썬 파일에서 (1) train시작 bash문 실행 + (2) train종료 bash문 종료(ctrl+c)

bash 문 실행할 파일이름을 start.sh 로 해서 생성해주기

import os
class TrainCallback(tf.keras.callbacks.Callback):
    def on_train_begin(self, logs=None):
        os.system("./start.sh &")
    def on_train_end(self, logs=None):
        os.system("mv start.txt " + file_name )
        os.system("pkill dcgmi") # start.sh 명령어 안에 dcgmi 가 있어서 dcgmi 포함하는것 강제종료..

파이썬에서 model.fit( 블라블라 )  이부분에서 아래 한줄 추가해주면 끝

callbacks = TrainCallback())

 

 

 

 

 

 

 

참조 :

https://www.tensorflow.org/api_docs/python/tf/keras/callbacks/Callback#on_train_begin

https://stackoverflow.com/questions/5789642/how-to-send-controlc-from-a-bash-script