basecls.zoo.benchmark#

basecls.zoo.benchmark.main()[源代码]#
basecls.zoo.benchmark.worker(args)[源代码]#
class basecls.zoo.benchmark.ClsBench(model, dataloader, trace=False)[源代码]#

基类:object

benchmark(warm_iters=50, log_seconds=2)[源代码]#
model_step(samples, targets)[源代码]#
class basecls.zoo.benchmark.TrainBench(model, dataloader, trace=False, amp_version=0)[源代码]#

基类:ClsBench

model_step(samples, targets)[源代码]#
class basecls.zoo.benchmark.EvalBench(model, dataloader, trace=False, amp_version=0)[源代码]#

基类:ClsBench

model_step(samples, targets)[源代码]#