pytorch像Keras一样的方式运行模型
from torchsample.modules import ModuleTrainer
class Network(nn.Module):
pass
model = Network()
trainer = ModuleTrainer(model)
trainer.compile(.......)
from torchsample.modules import ModuleTrainer
class Network(nn.Module):
pass
model = Network()
trainer = ModuleTrainer(model)
trainer.compile(.......)