force argument in print function

[iksooman]

Thanks for sharing your code!

I'm trying to test your code, but I got an error like:

Traceback (most recent call last):
  File "C:\Users\admin\PreNAS\supernet_train.py", line 463, in <module>
    main(args)
  File "C:\Users\admin\PreNAS\supernet_train.py", line 404, in main
    train_stats = train_one_epoch(
  File "C:\Users\admin\PreNAS\supernet_engine.py", line 187, in train_one_epoch
    print("DEBUG:train {}".format(config), force=print2file)
TypeError: 'force' is an invalid keyword argument for print()

Could you give me an idea for resolving this error?

[drcege]

Did you skip the distributed setup? The print function is replaced here:

PreNAS/lib/utils.py

Lines 165 to 177 in 0050c7a

	def setup_for_distributed(is_master):
	"""
	This function disables printing when not in master process
	"""
	import builtins as __builtin__
	builtin_print = __builtin__.print
	def print(*args, **kwargs):
	force = kwargs.pop('force', False)
	if is_master or force:
	builtin_print(*args, **kwargs)
	__builtin__.print = print