ColossalAI

46 Tags

236 MiB

Tree: 258b43317c

Author	SHA1	Message	Date
HELSON	5d3a2be3af	[amp] add gradient clipping for unit tests (#2283 ) * [amp] add gradient clipping in unit tests * fix bugs	2 years ago
HELSON	a1ce02d740	[zero] test gradient accumulation (#1964 ) * [zero] fix memory leak for zero2 * [zero] test gradient accumulation * [zero] remove grad clip test	2 years ago
Frank Lee	91cd34e6e0	[unittest] added doc for the pytest wrapper (#1704 )	2 years ago
Frank Lee	0e52f3d3d5	[unittest] supported condititonal testing based on env var (#1701 ) polish code	2 years ago
Super Daniel	8328917348	[NFC] polish colossalai/testing/comparison.py code style. (#1558 )	2 years ago
Frank Lee	50ec3a7e06	[test] skip tests when not enough GPUs are detected (#1090 ) * [test] skip tests when not enough GPUs are detected * polish code * polish code	2 years ago
Frank Lee	4ea49cb536	[test] added a decorator for address already in use error with backward compatibility (#760 ) * [test] added a decorator for address already in use error with backward compatibility * [test] added a decorator for address already in use error with backward compatibility	3 years ago
Frank Lee	3601b2bad0	[test] fixed rerun_on_exception and adapted test cases (#487 )	3 years ago
Frank Lee	83a847d058	[test] added rerun on exception for testing (#475 ) * [test] added rerun on exception function * polish code	3 years ago
Frank Lee	b72b8445c6	optimized context test time consumption (#446 )	3 years ago
Frank Lee	bffd85bf34	added testing module (#435 )	3 years ago