@ -3075,6 +3075,7 @@ def setup_args():
' training-default-halfp ' : False ,
' training-default-bnb ' : True ,
' training-oneapi ' : False ,
}
if os . path . isfile ( ' ./config/exec.json ' ) :
@ -3127,7 +3128,8 @@ def setup_args():
parser . add_argument ( " --training-default-halfp " , action = ' store_true ' , default = default_arguments [ ' training-default-halfp ' ] , help = " Training default: halfp " )
parser . add_argument ( " --training-default-bnb " , action = ' store_true ' , default = default_arguments [ ' training-default-bnb ' ] , help = " Training default: bnb " )
parser . add_argument ( " --training-oneapi " , action = ' store_true ' , default = default_arguments [ ' training-oneapi ' ] , help = " Train using oneAPI " )
parser . add_argument ( " --os " , default = " unix " , help = " Specifies which OS, easily " )
args = parser . parse_args ( )
@ -3156,6 +3158,15 @@ def setup_args():
args . listen_port = int ( args . listen_port )
if args . listen_port == 0 :
args . listen_port = None
if args . training_oneapi :
print ( " Training will happen with oneAPI. " ) # TODO: this gets printed twice. Find a better place to print it?
os . environ [ " AIVC_TRAIN_ONEAPI " ] = " one "
else :
try :
del os . environ [ " AIVC_TRAIN_ONEAPI " ]
except Exception as e :
pass
return args
@ -3200,6 +3211,7 @@ def get_default_settings( hypenated=True ):
' training-default-halfp ' : args . training_default_halfp ,
' training-default-bnb ' : args . training_default_bnb ,
' training-oneapi ' : args . training_oneapi ,
}
res = { }
@ -3252,6 +3264,7 @@ def update_args( **kwargs ):
args . training_default_halfp = settings [ ' training_default_halfp ' ]
args . training_default_bnb = settings [ ' training_default_bnb ' ]
args . training_oneapi = settings [ ' training_oneapi ' ]
save_args_settings ( )