fastNLP/tests/core/drivers/torch_driver
2022-05-15 05:45:33 +00:00
..
__init__.py 更新 callbacks, controllers, drivers, samplers 以及 tests 中的部分内容 2022-04-08 21:48:29 +08:00
test_ddp.py 1.修改torch测试例中多卡的driver参数 2.修改测试例中的driver.save driver.load为driver.save_checkpoint driver.load_checkpoint 3.添加lstm 2022-05-13 11:19:31 +00:00
test_dist_utils.py 调整测试例以适应pytest;为测试例添加标签 2022-05-02 05:46:47 +00:00
test_initialize_torch_driver.py 1.修改torch测试例中多卡的driver参数 2.修改测试例中的driver.save driver.load为driver.save_checkpoint driver.load_checkpoint 3.添加lstm 2022-05-13 11:19:31 +00:00
test_single_device.py paddle 和测试例跟进 set_dist_repro_dataloader 函数;修改test_trainer_wo_evaluator_torch.py的bug 2022-05-15 05:45:33 +00:00
test_torch_replace_sampler.py 增加RandomBatchSampler 2022-05-02 23:08:50 +08:00
test_utils.py 增加RandomBatchSampler 2022-05-02 23:08:50 +08:00