强制summary使用cpu计算

[cjld]

Jittor-summary/jittorsummary/jittorsummary.py

Line 13 in 3842c79

def summary(model, input_size, batch_size=-1, device=None, dtypes=None):

可以在这一行添加一个修饰符@jt.flag_scope(use_cuda=0)来强制使用cpu

[liuruiyang98]

将jt.int64更换为numpy.int64，现在jittorsummary已支持CUDA版本，通过 device=‘cpu’ 或者 device=‘cuda' 进行控制。