Pytorch "TypeError: add_record(): incompatible function arguments" while training the model
Created by: AndreiChirap
TypeError: add_record(): incompatible function arguments. The following argument types are supported: 1. (self: visualdl.core.HistogramWriter__float, arg0: int, arg1: List[float])
Here is the code : ###############################################################################
histogram
weight_list = net.conv1.weight.view(6*3*5*5,-1)
histogram0.add_record(train_step,weight_list)**<<<<------ here occurs the problem**
# image
image1.start_sampling()
image1.add_sample([96, 25], net.conv2.weight.view(16 * 6 * 5 * 5, -1))**<<<<------ and here**
image1.finish_sampling()
image2.start_sampling()
image2.add_sample([18, 25], net.conv1.weight.view(6 * 3 * 5 * 5, -1))**<<<<------ and here**
image2.finish_sampling()
################################################################################# also a similar problem occurs for "image1.add_sample([96, 25], net.conv2.weight.view(16 * 6 * 5 * 5, -1))"
Does anyone know what should I do ?